Skip to content

Commit f6eaba2

Browse files
authored
Flakey CI fix (pytorch#21110)
1 parent fcc1020 commit f6eaba2

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

examples/models/qwen3_5_moe/test_chunked_prefill.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -113,7 +113,10 @@ def test_chunked_prefill_matches_unchunked(self):
113113
logits_chunk.to(torch.float32),
114114
logits_full.to(torch.float32),
115115
rtol=1e-2,
116-
atol=1e-2,
116+
# cross-hardware MLX accumulation noise on near-zero logits (the
117+
# greedy-token assert above is the functional guard; a real chunk-
118+
# boundary desync would diverge by orders of magnitude more)
119+
atol=3e-2,
117120
)
118121

119122

0 commit comments

Comments
 (0)