Skip to content

GitLab

Unverified Commit 5767475a authored by Lei Wang's avatar Lei Wang Committed by GitHub
Browse files

[CI] Disable buggy(maybe) warp specialized kernel ci test for H20 (#1033)

parent eed320f5
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 6 deletions
examples/warp_specialize/test_example_warp_specialize.py
View file @ 5767475a
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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment