add ut test files test/functorch/test_ops.py and test/test_unary_ufuncs.py

Author: zxd1997066

test/xpu/functorch/common_utils.py

@@ -677,58 +677,7 @@
         # Accumulate error due to different accumulation order.
         "test_logcumsumexp_complex_xpu_complex64",
     ),
-    "test_unary_ufuncs_xpu.py": (
-        # AssertionError: Jiterator is only supported on CUDA and ROCm GPUs, none are available.
-        "_jiterator_",
-        # For extreme value processing, Numpy and XPU results are inconsistent
-        # std operations get different behavior on std::complex operarands for extremal cases
-        "test_reference_numerics_extremal__refs_log_xpu_complex64",
-        "test_reference_numerics_extremal_log_xpu_complex64",
-        "test_reference_numerics_extremal__refs_acos_xpu_complex64",
-        "test_reference_numerics_extremal__refs_acosh_xpu_complex64",
-        "test_reference_numerics_extremal_acos_xpu_complex64",
-        "test_reference_numerics_extremal_acosh_xpu_complex64",
-        "test_reference_numerics_extremal__refs_asinh_xpu_complex64",
-        "test_reference_numerics_extremal_asinh_xpu_complex64",
-        "test_reference_numerics_extremal__refs_asin_xpu_complex64",
-        "test_reference_numerics_extremal_asin_xpu_complex64",
-        "test_reference_numerics_large__refs_acosh_xpu_complex64",
-        "test_reference_numerics_large_acosh_xpu_complex64",
-        "test_reference_numerics_extremal__refs_log10_xpu_complex64",
-        "test_reference_numerics_extremal__refs_log1p_xpu_complex64",
-        "test_reference_numerics_extremal_log10_xpu_complex64",
-        "test_reference_numerics_extremal_log1p_xpu_complex64",
-        "test_reference_numerics_large__refs_asinh_xpu_complex128",
-        "test_reference_numerics_large__refs_asinh_xpu_complex64",
-        "test_reference_numerics_large__refs_asinh_xpu_complex32",
-        "test_reference_numerics_large_asinh_xpu_complex128",
-        "test_reference_numerics_large_asinh_xpu_complex64",
-        "test_reference_numerics_large_asinh_xpu_complex32",
-        # AssertionError: Tensor-likes are not close!
-        # exceeded maximum allowed difference
-        # Greatest absolute difference: 6.266784475883469e-05 at index (463, 204) (up to 1e-05 allowed)
-        # Greatest relative difference: 1.9145216356264427e-05 at index (463, 204) (up to 1.3e-06 allowed)
-        "test_reference_numerics_normal__refs_asinh_xpu_complex64",
-        "test_reference_numerics_normal_asinh_xpu_complex64",
-        # Unexpected success: CUDA uses thrust::sqrt and has accuracy issue. XPU use std::sqrt and has no issue.
-        "test_reference_numerics_large_rsqrt_xpu_complex32",
-        # Numeric difference
-        # https://github.com/intel/torch-xpu-ops/issues/544
-        # Expected 0.00497517 but got 0.00497520063072443.
-        # Absolute difference: 3.063072442997111e-08 (up to 0.0 allowed)
-        # Relative difference: 6.156719153309558e-06 (up to 1e-06 allowed)
-        "test_log1p_complex_xpu_complex64",
-        # Issue: https://github.com/intel/torch-xpu-ops/issues/622
-        # Mismatched elements: 8 / 943593 (0.0%)
-        # Greatest absolute difference: inf at index (9, 860) (up to 0.001 allowed)
-        # Greatest relative difference: inf at index (9, 860) (up to 0.0012 allowed)
-        "test_reference_numerics_normal_polygamma_polygamma_n_1_xpu_float16",
-        "test_reference_numerics_normal_polygamma_polygamma_n_2_xpu_float16",
-        "test_reference_numerics_normal_polygamma_polygamma_n_3_xpu_float16",
-        "test_reference_numerics_normal_polygamma_polygamma_n_4_xpu_float16",
-        # CUDA XFAIL
-        "test_reference_numerics_large__refs_rsqrt_xpu_complex32",
-    ),
+    "test_unary_ufuncs_xpu.py": None,
     "test_masked_xpu.py": None,
     "test_view_ops_xpu.py": (
         # Need quantization support, NotImplementedError: Could not run 'aten::_empty_affine_quantized' with arguments from the 'QuantizedXPU' backend.
@@ -1924,4 +1873,5 @@
         "test_sparse_matmul_xpu_float64",  # - RuntimeError: Double and complex datatype matmul is not supported in oneDNN
         "test_sparse_mm_xpu_float64",  # - NotImplementedError: Could not run 'aten::addmm' with arguments from the 'SparseXPU' backend. This could be because the operator doesn't exist for this backend, or wa...
     ),
+    "functorch/test_ops_functorch_xpu.py": None,
 }

test/xpu/functorch/functorch_additional_op_db.py

@@ -25,47 +25,6 @@
         "test_copy_from_dlpack_xpu_uint8",
     ),
     # SYCL compiler issue where host and device results differ for math ops with complex dtypes
-    "test_unary_ufuncs_xpu.py": (
-        "test_reference_numerics_extremal__refs_atanh_xpu_complex128",
-        "test_reference_numerics_extremal__refs_atanh_xpu_complex64",
-        "test_reference_numerics_extremal__refs_nn_functional_tanhshrink_xpu_complex128",
-        "test_reference_numerics_extremal__refs_sin_xpu_complex128",
-        "test_reference_numerics_extremal__refs_sin_xpu_complex64",
-        "test_reference_numerics_extremal__refs_sinh_xpu_complex128",
-        "test_reference_numerics_extremal__refs_sinh_xpu_complex64",
-        "test_reference_numerics_extremal__refs_tan_xpu_complex128",
-        "test_reference_numerics_extremal__refs_tan_xpu_complex64",
-        "test_reference_numerics_extremal_atanh_xpu_complex128",
-        "test_reference_numerics_extremal_atanh_xpu_complex64",
-        "test_reference_numerics_extremal_nn_functional_tanhshrink_xpu_complex128",
-        "test_reference_numerics_extremal_sin_xpu_complex128",
-        "test_reference_numerics_extremal_sin_xpu_complex64",
-        "test_reference_numerics_extremal_sinh_xpu_complex128",
-        "test_reference_numerics_extremal_sinh_xpu_complex64",
-        "test_reference_numerics_extremal_square_xpu_complex128",
-        "test_reference_numerics_extremal_square_xpu_complex64",
-        "test_reference_numerics_extremal_tan_xpu_complex128",
-        "test_reference_numerics_extremal_tan_xpu_complex64",
-        "test_reference_numerics_large__refs_cos_xpu_complex128",
-        "test_reference_numerics_large__refs_cos_xpu_complex32",
-        "test_reference_numerics_large__refs_cos_xpu_complex64",
-        "test_reference_numerics_large__refs_cosh_xpu_complex32",
-        "test_reference_numerics_large__refs_exp_xpu_complex128",
-        "test_reference_numerics_large__refs_exp_xpu_complex32",
-        "test_reference_numerics_large__refs_exp_xpu_complex64",
-        "test_reference_numerics_large__refs_sin_xpu_complex128",
-        "test_reference_numerics_large__refs_sin_xpu_complex32",
-        "test_reference_numerics_large__refs_sin_xpu_complex64",
-        "test_reference_numerics_large__refs_sinh_xpu_complex32",
-        "test_reference_numerics_large__refs_tan_xpu_complex32",
-        "test_reference_numerics_large_cos_xpu_complex128",
-        "test_reference_numerics_large_cos_xpu_complex32",
-        "test_reference_numerics_large_cos_xpu_complex64",
-        "test_reference_numerics_large_exp_xpu_complex128",
-        "test_reference_numerics_large_exp_xpu_complex64",
-        "test_reference_numerics_large_sin_xpu_complex128",
-        "test_reference_numerics_large_sin_xpu_complex32",
-        "test_reference_numerics_large_sin_xpu_complex64",
-        "test_reference_numerics_small_acos_xpu_complex32",
-    ),
+    "test_unary_ufuncs_xpu.py": None,
+    "functorch/test_ops_functorch_xpu.py": None,
 }

test/xpu/functorch/test_ops_functorch_xpu.py

@@ -32,148 +32,6 @@
         "test_non_contig_pow_xpu_complex64",
     ),
     "test_nn_xpu.py": ("test_adaptiveavg_pool1d_shmem_xpu",),
-    "test_unary_ufuncs_xpu.py": (
-        "test_batch_vs_slicing__refs_acos_xpu_complex64",
-        "test_batch_vs_slicing__refs_acosh_xpu_complex64",
-        "test_batch_vs_slicing__refs_log_xpu_complex64",
-        "test_batch_vs_slicing__refs_sqrt_xpu_complex64",
-        "test_batch_vs_slicing_acos_xpu_complex32",
-        "test_batch_vs_slicing_acos_xpu_complex64",
-        "test_batch_vs_slicing_acosh_xpu_complex32",
-        "test_batch_vs_slicing_acosh_xpu_complex64",
-        "test_batch_vs_slicing_log_xpu_complex32",
-        "test_batch_vs_slicing_log_xpu_complex64",
-        "test_batch_vs_slicing_sqrt_xpu_complex32",
-        "test_batch_vs_slicing_sqrt_xpu_complex64",
-        "test_batch_vs_slicing_square_xpu_complex64",
-        "test_contig_size1__refs_acos_xpu_complex64",
-        "test_contig_size1__refs_acosh_xpu_complex64",
-        "test_contig_size1__refs_log_xpu_complex64",
-        "test_contig_size1__refs_sqrt_xpu_complex64",
-        "test_contig_size1_acos_xpu_complex32",
-        "test_contig_size1_acos_xpu_complex64",
-        "test_contig_size1_acosh_xpu_complex32",
-        "test_contig_size1_acosh_xpu_complex64",
-        "test_contig_size1_large_dim__refs_acos_xpu_complex64",
-        "test_contig_size1_large_dim__refs_acosh_xpu_complex64",
-        "test_contig_size1_large_dim__refs_log_xpu_complex64",
-        "test_contig_size1_large_dim__refs_sqrt_xpu_complex64",
-        "test_contig_size1_large_dim_acos_xpu_complex32",
-        "test_contig_size1_large_dim_acos_xpu_complex64",
-        "test_contig_size1_large_dim_acosh_xpu_complex32",
-        "test_contig_size1_large_dim_acosh_xpu_complex64",
-        "test_contig_size1_large_dim_log_xpu_complex32",
-        "test_contig_size1_large_dim_log_xpu_complex64",
-        "test_contig_size1_large_dim_sqrt_xpu_complex32",
-        "test_contig_size1_large_dim_sqrt_xpu_complex64",
-        "test_contig_size1_large_dim_square_xpu_complex64",
-        "test_contig_size1_log_xpu_complex32",
-        "test_contig_size1_log_xpu_complex64",
-        "test_contig_size1_sqrt_xpu_complex32",
-        "test_contig_size1_sqrt_xpu_complex64",
-        "test_contig_size1_square_xpu_complex64",
-        "test_contig_vs_every_other__refs_acos_xpu_complex64",
-        "test_contig_vs_every_other__refs_acosh_xpu_complex64",
-        "test_contig_vs_every_other__refs_log_xpu_complex64",
-        "test_contig_vs_every_other__refs_sqrt_xpu_complex64",
-        "test_contig_vs_every_other_acos_xpu_complex32",
-        "test_contig_vs_every_other_acos_xpu_complex64",
-        "test_contig_vs_every_other_acosh_xpu_complex32",
-        "test_contig_vs_every_other_acosh_xpu_complex64",
-        "test_contig_vs_every_other_log_xpu_complex32",
-        "test_contig_vs_every_other_log_xpu_complex64",
-        "test_contig_vs_every_other_sqrt_xpu_complex32",
-        "test_contig_vs_every_other_sqrt_xpu_complex64",
-        "test_contig_vs_every_other_square_xpu_complex64",
-        "test_contig_vs_transposed__refs_acos_xpu_complex64",
-        "test_contig_vs_transposed__refs_acosh_xpu_complex64",
-        "test_contig_vs_transposed__refs_log_xpu_complex64",
-        "test_contig_vs_transposed__refs_sqrt_xpu_complex64",
-        "test_contig_vs_transposed_acos_xpu_complex32",
-        "test_contig_vs_transposed_acos_xpu_complex64",
-        "test_contig_vs_transposed_acosh_xpu_complex32",
-        "test_contig_vs_transposed_acosh_xpu_complex64",
-        "test_contig_vs_transposed_log_xpu_complex32",
-        "test_contig_vs_transposed_log_xpu_complex64",
-        "test_contig_vs_transposed_sqrt_xpu_complex32",
-        "test_contig_vs_transposed_sqrt_xpu_complex64",
-        "test_contig_vs_transposed_square_xpu_complex64",
-        "test_non_contig__refs_acos_xpu_complex64",
-        "test_non_contig__refs_acosh_xpu_complex64",
-        "test_non_contig__refs_log_xpu_complex64",
-        "test_non_contig__refs_sqrt_xpu_complex64",
-        "test_non_contig_acos_xpu_complex32",
-        "test_non_contig_acos_xpu_complex64",
-        "test_non_contig_acosh_xpu_complex32",
-        "test_non_contig_acosh_xpu_complex64",
-        "test_non_contig_expand__refs_acos_xpu_complex64",
-        "test_non_contig_expand__refs_acosh_xpu_complex64",
-        "test_non_contig_expand__refs_log_xpu_complex64",
-        "test_non_contig_expand__refs_sqrt_xpu_complex64",
-        "test_non_contig_expand_acos_xpu_complex32",
-        "test_non_contig_expand_acos_xpu_complex64",
-        "test_non_contig_expand_acosh_xpu_complex32",
-        "test_non_contig_expand_acosh_xpu_complex64",
-        "test_non_contig_expand_log_xpu_complex32",
-        "test_non_contig_expand_log_xpu_complex64",
-        "test_non_contig_expand_sqrt_xpu_complex32",
-        "test_non_contig_expand_sqrt_xpu_complex64",
-        "test_non_contig_expand_square_xpu_complex64",
-        "test_non_contig_index__refs_acos_xpu_complex64",
-        "test_non_contig_index__refs_acosh_xpu_complex64",
-        "test_non_contig_index__refs_log_xpu_complex64",
-        "test_non_contig_index__refs_sqrt_xpu_complex64",
-        "test_non_contig_index_acos_xpu_complex32",
-        "test_non_contig_index_acos_xpu_complex64",
-        "test_non_contig_index_acosh_xpu_complex32",
-        "test_non_contig_index_acosh_xpu_complex64",
-        "test_non_contig_index_log_xpu_complex32",
-        "test_non_contig_index_log_xpu_complex64",
-        "test_non_contig_index_sqrt_xpu_complex32",
-        "test_non_contig_index_sqrt_xpu_complex64",
-        "test_non_contig_index_square_xpu_complex64",
-        "test_non_contig_log_xpu_complex32",
-        "test_non_contig_log_xpu_complex64",
-        "test_non_contig_sqrt_xpu_complex32",
-        "test_non_contig_sqrt_xpu_complex64",
-        "test_non_contig_square_xpu_complex64",
-        "test_reference_numerics_extremal__refs_sqrt_xpu_complex64",
-        "test_reference_numerics_extremal_sqrt_xpu_complex64",
-        "test_reference_numerics_large__refs_acos_xpu_complex64",
-        "test_reference_numerics_large__refs_log_xpu_complex64",
-        "test_reference_numerics_large__refs_sqrt_xpu_complex64",
-        "test_reference_numerics_large_acos_xpu_complex32",
-        "test_reference_numerics_large_acos_xpu_complex64",
-        "test_reference_numerics_large_acosh_xpu_complex32",
-        "test_reference_numerics_large_log_xpu_complex32",
-        "test_reference_numerics_large_log_xpu_complex64",
-        "test_reference_numerics_large_sqrt_xpu_complex32",
-        "test_reference_numerics_large_sqrt_xpu_complex64",
-        "test_reference_numerics_normal__refs_acos_xpu_complex64",
-        "test_reference_numerics_normal__refs_acosh_xpu_complex64",
-        "test_reference_numerics_normal__refs_log_xpu_complex64",
-        "test_reference_numerics_normal__refs_sqrt_xpu_complex64",
-        "test_reference_numerics_normal_acos_xpu_complex32",
-        "test_reference_numerics_normal_acos_xpu_complex64",
-        "test_reference_numerics_normal_acosh_xpu_complex32",
-        "test_reference_numerics_normal_acosh_xpu_complex64",
-        "test_reference_numerics_normal_log_xpu_complex32",
-        "test_reference_numerics_normal_log_xpu_complex64",
-        "test_reference_numerics_normal_sqrt_xpu_complex32",
-        "test_reference_numerics_normal_sqrt_xpu_complex64",
-        "test_reference_numerics_normal_square_xpu_complex64",
-        "test_reference_numerics_small__refs_acos_xpu_complex64",
-        "test_reference_numerics_small__refs_acosh_xpu_complex64",
-        "test_reference_numerics_small__refs_log_xpu_complex64",
-        "test_reference_numerics_small__refs_sqrt_xpu_complex64",
-        "test_reference_numerics_small_acos_xpu_complex32",
-        "test_reference_numerics_small_acos_xpu_complex64",
-        "test_reference_numerics_small_acosh_xpu_complex32",
-        "test_reference_numerics_small_acosh_xpu_complex64",
-        "test_reference_numerics_small_log_xpu_complex32",
-        "test_reference_numerics_small_log_xpu_complex64",
-        "test_reference_numerics_small_sqrt_xpu_complex32",
-        "test_reference_numerics_small_sqrt_xpu_complex64",
-        "test_reference_numerics_small_square_xpu_complex64",
-    ),
+    "test_unary_ufuncs_xpu.py": None,
+    "functorch/test_ops_functorch_xpu.py": None,
 }

test/xpu/skip_list_common.py

@@ -1,31 +1,6 @@
 skip_dict = {
     # tensor(0.-0.j, device='xpu:0', dtype=torch.complex32) tensor(nan+nanj, device='xpu:0', dtype=torch.complex32) (1.5707964+0j)
-    "test_unary_ufuncs_xpu.pyy": (
-        "test_reference_numerics_small_acos_xpu_complex32",
-        "test_reference_numerics_small_asin_xpu_complex32",
-        "test_reference_numerics_small_asinh_xpu_complex32",
-        "test_reference_numerics_small_atan_xpu_complex32",
-        "test_reference_numerics_small_atanh_xpu_complex32",
-        # Need to check compiler std::sin() on inf+infj
-        "test_reference_numerics_extremal__refs_sin_xpu_complex128",
-        "test_reference_numerics_extremal__refs_sin_xpu_complex64",
-        "test_reference_numerics_extremal_nn_functional_tanhshrink_xpu_complex128",
-        "test_reference_numerics_extremal_nn_functional_tanhshrink_xpu_complex64",
-        "test_reference_numerics_extremal_sin_xpu_complex128",
-        "test_reference_numerics_extremal_sin_xpu_complex64",
-        "test_reference_numerics_extremal_sinh_xpu_complex128",
-        "test_reference_numerics_extremal_sinh_xpu_complex64",
-        "test_reference_numerics_large__refs_sin_xpu_complex32",
-        "test_reference_numerics_large_sin_xpu_complex32",
-        # Known issue of exp accuracy
-        # tensor(13437.7000-501.j, device='xpu:0', dtype=torch.complex128) tensor(inf+infj, device='xpu:0', dtype=torch.complex128) (-inf+infj)
-        "test_reference_numerics_large__refs_exp_xpu_complex128",
-        "test_reference_numerics_large_exp_xpu_complex128",
-        "test_reference_numerics_small_exp_xpu_complex32",
-        ":test_reference_numerics_normal_special_i1_xpu_float32",
-        "test_reference_numerics_normal_sigmoid_xpu_complex32",
-        "test_reference_numerics_small_sigmoid_xpu_complex32",
-    ),
+    "test_unary_ufuncs_xpu.py": None,
     # https://github.com/intel/torch-xpu-ops/issues/1171
     # AssertionError: 'Assertion maxind >= 0 && maxind < outputImageSize failed' not found in '\nAssertHandler::printMessage\n' : The expected error was not found
     "nn\test_pooling_xpu.py": (
@@ -36,4 +11,5 @@
         "test_MaxUnpool_index_errors_case7_xpu",
         "test_MaxUnpool_index_errors_case9_xpu",
     ),
+    "functorch/test_ops_functorch_xpu.py": None,
 }

test/xpu/skip_list_win.py

@@ -1,30 +1,6 @@
 skip_dict = {
     # tensor(0.-0.j, device='xpu:0', dtype=torch.complex32) tensor(nan+nanj, device='xpu:0', dtype=torch.complex32) (1.5707964+0j)
-    "test_unary_ufuncs_xpu.pyy": (
-        "test_reference_numerics_small_acos_xpu_complex32",
-        "test_reference_numerics_small_asin_xpu_complex32",
-        "test_reference_numerics_small_asinh_xpu_complex32",
-        "test_reference_numerics_small_atan_xpu_complex32",
-        "test_reference_numerics_small_atanh_xpu_complex32",
-        # Need to check compiler std::sin() on inf+infj
-        "test_reference_numerics_extremal__refs_sin_xpu_complex128",
-        "test_reference_numerics_extremal__refs_sin_xpu_complex64",
-        "test_reference_numerics_extremal_nn_functional_tanhshrink_xpu_complex128",
-        "test_reference_numerics_extremal_nn_functional_tanhshrink_xpu_complex64",
-        "test_reference_numerics_extremal_sin_xpu_complex128",
-        "test_reference_numerics_extremal_sin_xpu_complex64",
-        "test_reference_numerics_extremal_sinh_xpu_complex128",
-        "test_reference_numerics_extremal_sinh_xpu_complex64",
-        "test_reference_numerics_large__refs_sin_xpu_complex32",
-        "test_reference_numerics_large_sin_xpu_complex32",
-        # Known issue of exp accuracy
-        # tensor(13437.7000-501.j, device='xpu:0', dtype=torch.complex128) tensor(inf+infj, device='xpu:0', dtype=torch.complex128) (-inf+infj)
-        "test_reference_numerics_large__refs_exp_xpu_complex128",
-        "test_reference_numerics_large_exp_xpu_complex128",
-        "test_reference_numerics_small_exp_xpu_complex32",
-        ":test_reference_numerics_normal_special_i1_xpu_float32",
-        "test_reference_numerics_normal_sigmoid_xpu_complex32",
-    ),
+    "test_unary_ufuncs_xpu.py": None,
     # https://github.com/intel/torch-xpu-ops/issues/1171
     # AssertionError: 'Assertion maxind >= 0 && maxind < outputImageSize failed' not found in '\nAssertHandler::printMessage\n' : The expected error was not found
     "nn\test_pooling_xpu.py": (
@@ -35,4 +11,5 @@
         "test_MaxUnpool_index_errors_case7_xpu",
         "test_MaxUnpool_index_errors_case9_xpu",
     ),
+    "functorch/test_ops_functorch_xpu.py": None,
 }