@@ -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