File tree 1 file changed +2
-6
lines changed
onnxscript/tests/function_libs/torch_lib 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -1691,9 +1691,7 @@ def _where_input_wrangler(
1691
1691
"nn.functional.scaled_dot_product_attention" ,
1692
1692
nn_ops .aten_scaled_dot_product_attention ,
1693
1693
trace_only = True ,
1694
- )
1695
- .skip (
1696
- reason = "fixme: ORT crashes on Windows, segfaults randomly on Linux" ,
1694
+ tolerance = {torch .float32 : (1e-5 , 1e-5 )},
1697
1695
)
1698
1696
.skip (
1699
1697
matcher = lambda sample : (attn_mask := sample .kwargs .get ("attn_mask" )) is not None
@@ -1708,9 +1706,7 @@ def _where_input_wrangler(
1708
1706
"nn.functional.scaled_dot_product_attention_bool_mask" ,
1709
1707
nn_ops .aten_scaled_dot_product_attention_bool_mask ,
1710
1708
trace_only = True ,
1711
- )
1712
- .skip (
1713
- reason = "fixme: ORT crashes on Windows, segfaults randomly on Linux" ,
1709
+ tolerance = {torch .float32 : (1e-5 , 1e-5 )},
1714
1710
)
1715
1711
.skip (
1716
1712
matcher = lambda sample : (attn_mask := sample .kwargs .get ("attn_mask" )) is not None
You can’t perform that action at this time.
0 commit comments