diff --git a/examples/warp_specialize/test_example_warp_specialize.py b/examples/warp_specialize/test_example_warp_specialize.py index 0fee266a0..dee507790 100644 --- a/examples/warp_specialize/test_example_warp_specialize.py +++ b/examples/warp_specialize/test_example_warp_specialize.py @@ -1,16 +1,17 @@ import tilelang.testing -import example_warp_specialize_flashmla import example_warp_specialize_gemm_barrierpipe_stage2 import example_warp_specialize_gemm_copy_0_gemm_1 import example_warp_specialize_gemm_copy_1_gemm_0 import example_warp_specialize_gemm_softpipe_stage2 - -@tilelang.testing.requires_cuda -@tilelang.testing.requires_cuda_compute_version_eq(9, 0) -def test_example_warp_specialize_flashmla(): - example_warp_specialize_flashmla.main() +# TODO: skip for now as non-deterministic on H20 +# CC @cunxiao +# @tilelang.testing.requires_cuda +# @tilelang.testing.requires_cuda_compute_version_eq(9, 0) +# def test_example_warp_specialize_flashmla(): +# import example_warp_specialize_flashmla +# example_warp_specialize_flashmla.main() @tilelang.testing.requires_cuda