Skip to content

Commit fc3efec

Browse files
Pavle Josipovicclaude
andcommitted
Revert test_flash_mla.py: restore watcher skip for Blackhole
The watcher skip for issue #37631 was prematurely removed. Restore it until the underlying issue is fully resolved. Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent a65fa03 commit fc3efec

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

models/demos/deepseek_v3_b1/tests/unit_tests/test_flash_mla.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
from loguru import logger
1313

1414
import ttnn
15-
from models.common.utility_functions import comp_pcc
15+
from models.common.utility_functions import comp_pcc, is_blackhole, is_watcher_enabled
1616
from models.demos.deepseek_v3_b1.micro_ops.flash_mla.op import FlashMLADecode
1717

1818

@@ -22,6 +22,8 @@
2222
@pytest.mark.parametrize("max_seq_len", [32 * 1024]) # 32k max sequence length per chip
2323
def test_flash_mla_decode(device, batch_size, num_chunks, k_chunk_size, max_seq_len):
2424
"""Test FlashMLADecode op."""
25+
if is_blackhole() and is_watcher_enabled():
26+
pytest.skip("Skipping test on Blackhole with watcher enabled, see issue #37631")
2527

2628
# Calculate decode_position from num_chunks and k_chunk_size
2729
decode_position = num_chunks * k_chunk_size - 1

0 commit comments

Comments
 (0)