diff --git a/tests/attention/test_fmha_v2_prefill.py b/tests/attention/test_fmha_v2_prefill.py index 90d075c649..cb9f2f3fbe 100644 --- a/tests/attention/test_fmha_v2_prefill.py +++ b/tests/attention/test_fmha_v2_prefill.py @@ -3,9 +3,9 @@ import math from typing import Optional, Tuple, Union -pytestmark = pytest.mark.skip( - reason="todo(jimmyzho): temporarily skip this test due to hangs" -) +# pytestmark = pytest.mark.skip( +# reason="todo(jimmyzho): temporarily skip this test due to hangs" +# ) import flashinfer from flashinfer.prefill import fmha_v2_prefill_deepseek @@ -843,7 +843,7 @@ def test_trtllm_fmha_v2_prefill( pytest.skip("Skip due to bug in fp8 sliding window") if mask_mode == "SLIDING_WINDOW": pytest.skip("todo(jimmyzho): temporarily skip sliding window test due to hang") - if dtype == torch.float8_e4m3fn and o_dtype == torch.float8_e4m3fn: + if dtype == torch.float8_e4m3fn: pytest.skip("todo(jimmyzho): temporarily skip fp8 tests due to hang") run_trtllm_fmha_v2_prefill_case( input_layout=input_layout,