fix

intel · Nov 12, 2024 · bed5a45 · bed5a45
1 parent 199b4ce
commit bed5a45
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/test/xpu/skip_list_common.py b/test/xpu/skip_list_common.py
@@ -810,7 +810,7 @@
         # https://github.com/intel/torch-xpu-ops/issues/761
         # AssertionError: False is not true
         # CPU fallback failure. To support aten::transformer_encoder_layer_forward with proper priority.
-        # "test_disable_fastpath_xpu",
+        "test_disable_fastpath_xpu",
         # We have no mechanism to handle SDPBackend::ERROR so far. Will give a fully support when we support all SDPBackends.
         "test_dispatch_fails_no_backend_xpu",
         # Could not run 'aten::_to_copy' with arguments from the 'NestedTensorXPU' backend

diff --git a/test/xpu/xpu_test_utils.py b/test/xpu/xpu_test_utils.py
@@ -218,6 +218,7 @@
     "nn.functional.ctc_loss",
     "nn.functional.channel_shuffle",
     "nn.functional.multi_head_attention_forward",
+    "nn.TransformerEncoderLayer",
     "sigmoid",
     "logsigmoid",
     "sgn",