test_example_gemm_sp.py 266 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
import tilelang.testing

import example_custom_compress
import example_gemm_sp


def test_example_custom_compress():
    example_custom_compress.main()


def test_example_gemm_sp():
    example_gemm_sp.main()


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