test_tilelang_intimm.py 517 Bytes
Newer Older
1
2
3
4
5
6
import tilelang
import tilelang.testing
import tilelang.language as T


def test_tilelang_intimm():
7
8
9
10
11
12
    T.int32(0x7FFFFFFF)
    T.int32(-0x7FFFFFFF - 1)
    T.uint32(0xFFFFFFFF)
    T.int64(0x7FFFFFFFFFFFFFFF)
    T.int64(-0x7FFFFFFFFFFFFFFF - 1)
    T.uint64(0xFFFFFFFFFFFFFFFF)
13
14

    a = T.int32()
15
    a & 0x7FFFFFFF
16
17

    a = T.uint32()
18
    a & 0xFFFFFFFF
19
20

    a = T.int64()
21
    a & 0x7FFFFFFFFFFFFFFF
22
23

    a = T.uint64()
24
    a & T.uint64(0xFFFFFFFFFFFFFFFF)
25
26


27
if __name__ == "__main__":
28
    tilelang.testing.main()