"src/vscode:/vscode.git/clone" did not exist on "485245a00699a367708a778617975c5258958292"
test_example_gemm.py 504 Bytes
Newer Older
1
2
3
4
5
6
7
8
import tilelang.testing
import example_gemm_autotune
import example_gemm_intrinsics
import example_gemm_schedule
import example_gemm


def test_example_gemm_autotune():
9
10
    # enable roller for fast tuning
    example_gemm_autotune.main(with_roller=True)
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26


def test_example_gemm_intrinsics():
    example_gemm_intrinsics.main()


def test_example_gemm_schedule():
    example_gemm_schedule.main()


def test_example_gemm():
    example_gemm.main()


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