Add aten::special_{i0e, i1, i1e, ndtri, log_ndtr, polynomial} OPs

LuFinch commented 1 month ago

[x] special_hermite_polynomial_he
[x] special_hermite_polynomial_h
[x] special_laguerre_polynomial_l
[x] special_legendre_polynomial_p
[x] special_i0e
[x] special_i1e
[x] special_i1
[x] special_log_ndtr
[x] special_ndtri

xytintel commented 4 weeks ago

@LuFinch Pls modify the filename. (e.g. sycl/Hermite_polynomial_he.cpp -> sycl/HermitePolynomialHeKernel.cpp)

EikanWang commented 4 weeks ago

Pls. help provide test case evidence.

LuFinch commented 4 weeks ago

Copy part of log from pull /preci-ut/Torch-XPU-UT-Tests. All other cases releated to this PR can pass.

2024-10-24T06:11:11.3652902Z test_ops_xpu.py::TestCommonXPU::test_dtypes_special_hermite_polynomial_h_xpu PASSED [  8%]
2024-10-24T06:11:11.4028181Z test_ops_xpu.py::TestCommonXPU::test_dtypes_special_hermite_polynomial_he_xpu PASSED [  8%]
2024-10-24T06:11:11.4215663Z test_ops_xpu.py::TestCommonXPU::test_dtypes_special_i0e_xpu PASSED       [  8%]
2024-10-24T06:11:11.4406891Z test_ops_xpu.py::TestCommonXPU::test_dtypes_special_i1_xpu PASSED        [  8%]
2024-10-24T06:11:11.4590297Z test_ops_xpu.py::TestCommonXPU::test_dtypes_special_i1e_xpu PASSED       [  8%]
2024-10-24T06:11:11.4973412Z test_ops_xpu.py::TestCommonXPU::test_dtypes_special_laguerre_polynomial_l_xpu PASSED [  8%]
2024-10-24T06:11:11.4979375Z test_ops_xpu.py::TestCommonXPU::test_dtypes_special_legendre_polynomial_p_xpu SKIPPED [  8%]
2024-10-24T06:11:11.5316997Z test_ops_xpu.py::TestCommonXPU::test_dtypes_special_log_ndtr_xpu PASSED  [  8%]
2024-10-24T06:11:19.7434737Z test_ops_xpu.py::TestCommonXPU::test_non_standard_bool_values_special_hermite_polynomial_h_xpu_bool PASSED [ 18%]
2024-10-24T06:11:19.7523585Z test_ops_xpu.py::TestCommonXPU::test_non_standard_bool_values_special_hermite_polynomial_he_xpu_bool PASSED [ 18%]
2024-10-24T06:11:19.7559269Z test_ops_xpu.py::TestCommonXPU::test_non_standard_bool_values_special_i0e_xpu_bool PASSED [ 18%]
2024-10-24T06:11:19.7599595Z test_ops_xpu.py::TestCommonXPU::test_non_standard_bool_values_special_i1_xpu_bool PASSED [ 18%]
2024-10-24T06:11:19.7635309Z test_ops_xpu.py::TestCommonXPU::test_non_standard_bool_values_special_i1e_xpu_bool PASSED [ 18%]
2024-10-24T06:11:19.7726125Z test_ops_xpu.py::TestCommonXPU::test_non_standard_bool_values_special_laguerre_polynomial_l_xpu_bool PASSED [ 18%]
2024-10-24T06:11:19.7733345Z test_ops_xpu.py::TestCommonXPU::test_non_standard_bool_values_special_legendre_polynomial_p_xpu_bool SKIPPED [ 18%]
2024-10-24T06:11:19.7774764Z test_ops_xpu.py::TestCommonXPU::test_non_standard_bool_values_special_log_ndtr_xpu_bool PASSED [ 18%]
2024-10-24T06:11:53.4370375Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_hermite_polynomial_h_xpu_float32 PASSED [ 26%]
2024-10-24T06:11:53.4451427Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_hermite_polynomial_h_xpu_int64 PASSED [ 26%]
2024-10-24T06:11:53.4535834Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_hermite_polynomial_he_xpu_float32 PASSED [ 26%]
2024-10-24T06:11:53.4615835Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_hermite_polynomial_he_xpu_int64 PASSED [ 26%]
2024-10-24T06:11:53.4671267Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_i0e_xpu_float32 PASSED [ 26%]
2024-10-24T06:11:53.4711989Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_i0e_xpu_int64 PASSED [ 26%]
2024-10-24T06:11:53.4778705Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_i1_xpu_float32 PASSED [ 26%]
2024-10-24T06:11:53.4816292Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_i1_xpu_int64 PASSED [ 26%]
2024-10-24T06:11:53.4881737Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_i1e_xpu_float32 PASSED [ 26%]
2024-10-24T06:11:53.4919326Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_i1e_xpu_int64 PASSED [ 26%]
2024-10-24T06:11:53.5004811Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_laguerre_polynomial_l_xpu_float32 PASSED [ 26%]
2024-10-24T06:11:53.5085358Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_laguerre_polynomial_l_xpu_int64 PASSED [ 26%]
2024-10-24T06:11:53.5092580Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_legendre_polynomial_p_xpu_float32 SKIPPED [ 26%]
2024-10-24T06:11:53.5099026Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_legendre_polynomial_p_xpu_int64 SKIPPED [ 26%]
2024-10-24T06:11:53.5162638Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_log_ndtr_xpu_float32 PASSED [ 26%]
2024-10-24T06:11:53.5204238Z test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_special_log_ndtr_xpu_int64 PASSED [ 26%]
2024-10-24T06:12:15.8440628Z test_ops_xpu.py::TestCommonXPU::test_out_special_hermite_polynomial_h_xpu_float32 PASSED [ 36%]
2024-10-24T06:12:15.8582718Z test_ops_xpu.py::TestCommonXPU::test_out_special_hermite_polynomial_he_xpu_float32 PASSED [ 36%]
2024-10-24T06:12:15.8631082Z test_ops_xpu.py::TestCommonXPU::test_out_special_i0e_xpu_float32 PASSED  [ 36%]
2024-10-24T06:12:15.8678355Z test_ops_xpu.py::TestCommonXPU::test_out_special_i1_xpu_float32 PASSED   [ 36%]
2024-10-24T06:12:15.8725716Z test_ops_xpu.py::TestCommonXPU::test_out_special_i1e_xpu_float32 PASSED  [ 36%]
2024-10-24T06:12:15.8870183Z test_ops_xpu.py::TestCommonXPU::test_out_special_laguerre_polynomial_l_xpu_float32 PASSED [ 36%]
2024-10-24T06:12:15.8875786Z test_ops_xpu.py::TestCommonXPU::test_out_special_legendre_polynomial_p_xpu_float32 SKIPPED [ 36%]
2024-10-24T06:12:15.8931645Z test_ops_xpu.py::TestCommonXPU::test_out_special_log_ndtr_xpu_float32 PASSED [ 36%]
2024-10-24T06:12:40.0638697Z test_ops_xpu.py::TestCommonXPU::test_out_warning_special_hermite_polynomial_h_xpu PASSED [ 41%]
2024-10-24T06:12:40.0909441Z test_ops_xpu.py::TestCommonXPU::test_out_warning_special_hermite_polynomial_he_xpu PASSED [ 41%]
2024-10-24T06:12:40.0997163Z test_ops_xpu.py::TestCommonXPU::test_out_warning_special_i0e_xpu PASSED  [ 41%]
2024-10-24T06:12:40.1079479Z test_ops_xpu.py::TestCommonXPU::test_out_warning_special_i1_xpu PASSED   [ 41%]
2024-10-24T06:12:40.1162678Z test_ops_xpu.py::TestCommonXPU::test_out_warning_special_i1e_xpu PASSED  [ 41%]
2024-10-24T06:12:40.1439267Z test_ops_xpu.py::TestCommonXPU::test_out_warning_special_laguerre_polynomial_l_xpu PASSED [ 41%]
2024-10-24T06:12:40.1444863Z test_ops_xpu.py::TestCommonXPU::test_out_warning_special_legendre_polynomial_p_xpu SKIPPED [ 41%]
2024-10-24T06:12:40.1533390Z test_ops_xpu.py::TestCommonXPU::test_out_warning_special_log_ndtr_xpu PASSED [ 41%]

intel / torch-xpu-ops

Add aten::special_{i0e, i1, i1e, ndtri, log_ndtr, polynomial} OPs #1014