test_example_gemm_splitk.py

import tilelang.testing

from example_tilelang_gemm_splitk import main


def test_example_tilelang_gemm_splitk():
    main()


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