search

intel / torch-xpu-ops

Apache License 2.0
30 stars 21 forks source link

Add aten::hardshrink and its variants #947

Closed xytintel closed 1 month ago

xytintel commented 1 month ago
xytintel commented 1 month ago

test_ops_xpu.py::TestCommonXPU::test_compare_cpurefs_nn_functional_hardshrink_xpu_float32 PASSED [ 0%] test_ops_xpu.py::TestCommonXPU::test_compare_cpu_nn_functional_hardshrink_xpu_float32 PASSED [ 2%] test_ops_xpu.py::TestCommonXPU::test_dtypes__refs_nn_functional_hardshrink_xpu PASSED [ 4%] test_ops_xpu.py::TestCommonXPU::test_dtypes_nn_functional_hardshrink_xpu PASSED [ 7%] test_ops_xpu.py::TestCommonXPU::test_noncontiguous_samples_nn_functional_hardshrink_xpu_float32 PASSED [ 25%] test_ops_xpu.py::TestCommonXPU::test_outrefs_nn_functional_hardshrink_xpu_float32 PASSED [ 30%] test_ops_xpu.py::TestCommonXPU::test_out_nn_functional_hardshrink_xpu_float32 PASSED [ 33%] test_ops_xpu.py::TestCommonXPU::test_out_requires_grad_error_nn_functional_hardshrink_xpu_float32 PASSED [ 35%] test_ops_xpu.py::TestCommonXPU::test_out_warning__refs_nn_functional_hardshrink_xpu PASSED [ 38%] test_ops_xpu.py::TestCommonXPU::test_out_warning_nn_functional_hardshrink_xpu PASSED [ 40%]