import tilelang.testing import example_custom_compress import example_gemm_sp def test_example_custom_compress(): example_custom_compress.main() def test_example_gemm_sp(): example_gemm_sp.main() if __name__ == "__main__": tilelang.testing.main()