test_example_tilelang_gemm_splitk.py

import tilelang.testing

from example_tilelang_gemm_streamk import main


def test_example_tilelang_gemm_streamk():
    main()


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