test_example_gemm.py

import tilelang.testing
import example_gemm_autotune
import example_gemm_intrinsics
import example_gemm_schedule
import example_gemm


def test_example_gemm_autotune():
    # enable roller for fast tuning
    example_gemm_autotune.main(M=1024, N=1024, K=1024, with_roller=True)


def test_example_gemm_intrinsics():
    example_gemm_intrinsics.main(M=1024, N=1024, K=1024)


def test_example_gemm_schedule():
    example_gemm_schedule.main()


def test_example_gemm():
    example_gemm.main()


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