scaled_dot_product_attention

Author: justinchuby

onnxscript/tests/function_libs/torch_lib/ops_test_data.py

@@ -1692,9 +1692,6 @@ def _where_input_wrangler(
         nn_ops.aten_scaled_dot_product_attention,
         trace_only=True,
     )
-    .skip(
-        reason="fixme: ORT crashes on Windows, segfaults randomly on Linux",
-    )
     .skip(
         matcher=lambda sample: (attn_mask := sample.kwargs.get("attn_mask")) is not None
         and attn_mask.dtype == torch.bool,
@@ -1709,9 +1706,6 @@ def _where_input_wrangler(
         nn_ops.aten_scaled_dot_product_attention_bool_mask,
         trace_only=True,
     )
-    .skip(
-        reason="fixme: ORT crashes on Windows, segfaults randomly on Linux",
-    )
     .skip(
         matcher=lambda sample: (attn_mask := sample.kwargs.get("attn_mask")) is not None
         and attn_mask.dtype != torch.bool,