test_example_gemm_fp8.py 459 Bytes
Newer Older
root's avatar
init  
root committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
import tilelang.testing
import example_tilelang_gemm_fp8_2xAcc
import example_tilelang_gemm_fp8_intrinsic
import example_tilelang_gemm_fp8


def test_example_tilelang_gemm_fp8_2xAcc():
    example_tilelang_gemm_fp8_2xAcc.main()


def test_example_tilelang_gemm_fp8_intrinsic():
    example_tilelang_gemm_fp8_intrinsic.main()


def test_example_tilelang_gemm_fp8():
    example_tilelang_gemm_fp8.main()


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