Skip to content

Commit 9e3eef7

Browse files
committed
Add ut test_decomp.py on XPU
1 parent 0bbb067 commit 9e3eef7

File tree

2 files changed

+1417
-1
lines changed

2 files changed

+1417
-1
lines changed

test/xpu/run_test_with_only.py

Lines changed: 53 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,59 @@ def launch_test(test_case, skip_list=None, exe_list=None):
4343
"test_comprehensive_nn_functional_nll_loss_xpu_float64",
4444
"bincount",
4545
)
46-
res += launch_test("test_decomp_xpu.py", exe_list=execute_list)
46+
skip_list = (
47+
"test_comprehensive_baddbmm_xpu_float64",
48+
"test_comprehensive_logspace_tensor_overload_xpu_int16",
49+
"test_comprehensive_logspace_tensor_overload_xpu_int32",
50+
"test_comprehensive_logspace_tensor_overload_xpu_int64",
51+
"test_comprehensive_logspace_xpu_int16",
52+
"test_comprehensive_logspace_xpu_int32",
53+
"test_comprehensive_logspace_xpu_int64",
54+
"test_comprehensive_nn_functional_conv_transpose2d_xpu_bfloat16",
55+
"test_comprehensive_nn_functional_conv_transpose2d_xpu_complex128",
56+
"test_comprehensive_nn_functional_conv_transpose2d_xpu_complex32",
57+
"test_comprehensive_nn_functional_conv_transpose2d_xpu_complex64",
58+
"test_comprehensive_nn_functional_conv_transpose2d_xpu_float16",
59+
"test_comprehensive_nn_functional_conv_transpose2d_xpu_float32",
60+
"test_comprehensive_nn_functional_conv_transpose2d_xpu_float64",
61+
"test_comprehensive_nn_functional_conv_transpose3d_xpu_bfloat16",
62+
"test_comprehensive_nn_functional_conv_transpose3d_xpu_complex128",
63+
"test_comprehensive_nn_functional_conv_transpose3d_xpu_complex32",
64+
"test_comprehensive_nn_functional_conv_transpose3d_xpu_complex64",
65+
"test_comprehensive_nn_functional_conv_transpose3d_xpu_float16",
66+
"test_comprehensive_nn_functional_conv_transpose3d_xpu_float32",
67+
"test_comprehensive_nn_functional_conv_transpose3d_xpu_float64",
68+
"test_comprehensive_nn_functional_instance_norm_xpu_float64",
69+
"test_comprehensive_nn_functional_nll_loss_xpu_float16",
70+
"test_comprehensive_nn_functional_pad_reflect_xpu_bfloat16",
71+
"test_comprehensive_torch_ops_aten__flash_attention_forward_xpu_float16",
72+
"test_comprehensive_vdot_xpu_complex128",
73+
"test_comprehensive_vdot_xpu_complex64",
74+
"test_quick_addmm_xpu_float64",
75+
"test_quick_baddbmm_xpu_float64",
76+
"test_quick_core_backward_baddbmm_xpu_float64",
77+
"test_quick_core_backward_mv_xpu_float64",
78+
"test_quick_logspace_tensor_overload_xpu_int16",
79+
"test_quick_logspace_tensor_overload_xpu_int32",
80+
"test_quick_logspace_tensor_overload_xpu_int64",
81+
"test_quick_logspace_xpu_int16",
82+
"test_quick_logspace_xpu_int32",
83+
"test_quick_logspace_xpu_int64",
84+
"test_quick_vdot_xpu_complex128",
85+
"test_quick_vdot_xpu_complex64",
86+
"test_exponential_non_inf_xpu",
87+
"test_aten_core_operators",
88+
"test_has_decomposition",
89+
"test_comprehensive_diff_xpu_complex128",
90+
"test_comprehensive_ormqr_xpu_complex128",
91+
"test_quick_var_mean_xpu_float64",
92+
"test_comprehensive_diff_xpu_complex64",
93+
"test_comprehensive_ormqr_xpu_complex64",
94+
"test_quick_mean_xpu_complex128",
95+
"test_comprehensive_grid_sampler_2d_xpu_bfloat16",
96+
)
97+
# res += launch_test("test_decomp_xpu.py", exe_list=execute_list)
98+
res += launch_test("test_decomp.py", skip_list=skip_list)
4799

48100
if os.name == "nt":
49101
sys.exit(res)

0 commit comments

Comments
 (0)