Skip to content

Commit 1c03efe

Browse files
committed
Temp skip batch prefill tests for causal/logits=False
Temporarily skip specific batch prefill test cases (causal=True, logits_soft_cap=0.0). This is due to a known issue with the ROCm 7.2 compiler in certain batch prefill kernel scenarios.
1 parent 9b542ad commit 1c03efe

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

op_tests/test_batch_prefill.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -726,6 +726,14 @@ def test_batch_prefill(
726726
is_input_fp8, dtype, causal, kv_len, qo_len, contiguous_kv
727727
):
728728
return {"status": "skipped"}
729+
730+
# Temporarily skip test for causal=True, logits_soft_cap=0.0
731+
if skip_test_if(
732+
causal and logits_soft_cap == 0.0,
733+
"Temporarily skip test for causal=True, logits_soft_cap=0.0",
734+
):
735+
return {"status": "skipped"}
736+
729737
if check_layout_skip_conditions(
730738
kvcache_layout,
731739
head_dim,

0 commit comments

Comments
 (0)