Skip to content

Commit

Permalink
Add MTL skip list for windows in extended folder
Browse files Browse the repository at this point in the history
Signed-off-by: Cheng, Penghui <[email protected]>
  • Loading branch information
PenghuiCheng committed Dec 24, 2024
1 parent 804fe26 commit 9730b8c
Show file tree
Hide file tree
Showing 2 changed files with 42 additions and 0 deletions.
22 changes: 22 additions & 0 deletions test/xpu/extended/run_test_with_skip_mtl.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
import os
import pytest
import sys
from skip_list_common import skip_dict
from skip_list_win import skip_dict as skip_dict_win
from skip_list_win_mtl import skip_dict as skip_dict_win_mtl

IS_WINDOWS = sys.platform == "win32"

skip_list = skip_dict["test_ops_xpu.py"]
if IS_WINDOWS:
skip_list += skip_dict_win["test_ops_xpu.py"] + skip_dict_win_mtl["test_ops_xpu.py"]

skip_options = "not " + skip_list[0]
for skip_case in skip_list[1:]:
skip_option = " and not " + skip_case
skip_options += skip_option

os.environ["PYTORCH_TEST_WITH_SLOW"]="1"
test_command = ["-k", skip_options, "test_ops_xpu.py", "-v"]
res = pytest.main(test_command)
sys.exit(res)
20 changes: 20 additions & 0 deletions test/xpu/extended/skip_list_win_mtl.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
skip_dict = {
# failed on MTL windows, skip first for Preci
"test_ops_xpu.py": (
"test_compare_cpu_sqrt_xpu_complex64",
"test_backward_nn_functional_adaptive_avg_pool2d_xpu_float32",

"test_compare_cpu_cosh_xpu_complex128",
"test_compare_cpu_frexp_xpu_bfloat16",
"test_compare_cpu_frexp_xpu_float16",
"test_compare_cpu_frexp_xpu_float32",
"test_compare_cpu_frexp_xpu_float64",
"test_compare_cpu_max_pool2d_with_indices_backward_xpu_bfloat16",
"test_compare_cpu_max_pool2d_with_indices_backward_xpu_float16",
"test_compare_cpu_max_pool2d_with_indices_backward_xpu_float32",
"test_compare_cpu_max_pool2d_with_indices_backward_xpu_float64",
"test_compare_cpu_nn_functional_avg_pool2d_xpu_bfloat16",
"test_compare_cpu_nn_functional_avg_pool2d_xpu_float32",
"test_compare_cpu_nn_functional_avg_pool3d_xpu_float32",
),
}

0 comments on commit 9730b8c

Please sign in to comment.