Skip to content

Commit f900c38

Browse files
authored
Qwen3 MoE test flakiness (apple#63)
1 parent 7af7755 commit f900c38

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

python/tests/test_model_units/test_models/test_macos_layers/test_qwen3_moe.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1351,8 +1351,8 @@ def test_qwen3_moe_layers(
13511351
backend=cast("Backend", Backend.coreai),
13521352
)
13531353

1354-
rtol = {Precision.f32: 1e-5, Precision.f16: 5e-2, Precision.bf16: 1e-1}[precision]
1355-
atol = {Precision.f32: 1e-5, Precision.f16: 5e-2, Precision.bf16: 1e-1}[precision]
1354+
rtol = {Precision.f32: 1e-5, Precision.f16: 5e-2, Precision.bf16: 2e-1}[precision]
1355+
atol = {Precision.f32: 1e-5, Precision.f16: 5e-2, Precision.bf16: 2e-1}[precision]
13561356
with tempfile.TemporaryDirectory() as temp_directory:
13571357
model = model_class(
13581358
Path(temp_directory),
@@ -1426,8 +1426,8 @@ def test_qwen3_moe_sparse_block(
14261426
backend=cast("Backend", Backend.coreai),
14271427
)
14281428

1429-
rtol = {Precision.f32: 1e-5, Precision.f16: 5e-2, Precision.bf16: 1e-1}[precision]
1430-
atol = {Precision.f32: 1e-5, Precision.f16: 5e-2, Precision.bf16: 1e-1}[precision]
1429+
rtol = {Precision.f32: 1e-5, Precision.f16: 5e-2, Precision.bf16: 2e-1}[precision]
1430+
atol = {Precision.f32: 1e-5, Precision.f16: 5e-2, Precision.bf16: 2e-1}[precision]
14311431

14321432
with tempfile.TemporaryDirectory() as temp_directory:
14331433
model = Qwen3MoeSparseMoeBlock(Path(temp_directory), top_k=top_k)

0 commit comments

Comments
 (0)