mirror of
https://github.com/tile-ai/tilelang.git
synced 2026-10-22 06:50:25 +08:00
Kuris
2426090fdb
* [Feat] add missing support of uint32x2 * [Feat] Add `T.Ref` annotation and tests * fix lint error * minor update for error message on twice decl * Remove unused let_bindings_ in CodeGenC to fix #1300
29 lines
517 B
Python
29 lines
517 B
Python
import tilelang
|
|
import tilelang.testing
|
|
import tilelang.language as T
|
|
|
|
|
|
def test_tilelang_intimm():
|
|
T.int32(0x7fffffff)
|
|
T.int32(-0x7fffffff - 1)
|
|
T.uint32(0xffffffff)
|
|
T.int64(0x7fffffffffffffff)
|
|
T.int64(-0x7fffffffffffffff - 1)
|
|
T.uint64(0xffffffffffffffff)
|
|
|
|
a = T.int32()
|
|
a & 0x7fffffff
|
|
|
|
a = T.uint32()
|
|
a & 0xffffffff
|
|
|
|
a = T.int64()
|
|
a & 0x7fffffffffffffff
|
|
|
|
a = T.uint64()
|
|
a & T.uint64(0xffffffffffffffff)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
tilelang.testing.main()
|