diff --git a/.github/workflows/conda-package.yml b/.github/workflows/conda-package.yml index 02e68ac8e82..fdf289a9630 100644 --- a/.github/workflows/conda-package.yml +++ b/.github/workflows/conda-package.yml @@ -18,6 +18,7 @@ env: test_linalg.py test_mathematical.py test_random_state.py + test_sort.py test_special.py test_usm_type.py VER_JSON_NAME: 'version.json' diff --git a/dpnp/backend/kernels/dpnp_krnl_elemwise.cpp b/dpnp/backend/kernels/dpnp_krnl_elemwise.cpp index 50440cdebc8..30310162582 100644 --- a/dpnp/backend/kernels/dpnp_krnl_elemwise.cpp +++ b/dpnp/backend/kernels/dpnp_krnl_elemwise.cpp @@ -811,8 +811,8 @@ static void func_map_init_elemwise_1arg_1type(func_map_t& fmap) fmap[DPNPFuncName::DPNP_FN_COPY_EXT][eft_LNG][eft_LNG] = {eft_LNG, (void*)dpnp_copy_c_ext}; fmap[DPNPFuncName::DPNP_FN_COPY_EXT][eft_FLT][eft_FLT] = {eft_FLT, (void*)dpnp_copy_c_ext}; fmap[DPNPFuncName::DPNP_FN_COPY_EXT][eft_DBL][eft_DBL] = {eft_DBL, (void*)dpnp_copy_c_ext}; - fmap[DPNPFuncName::DPNP_FN_COPY_EXT][eft_C128][eft_C128] = {eft_C128, - (void*)dpnp_copy_c_ext>}; + fmap[DPNPFuncName::DPNP_FN_COPY_EXT][eft_C64][eft_C64] = {eft_C64, (void*)dpnp_copy_c_ext>}; + fmap[DPNPFuncName::DPNP_FN_COPY_EXT][eft_C128][eft_C128] = {eft_C128, (void*)dpnp_copy_c_ext>}; fmap[DPNPFuncName::DPNP_FN_ERF][eft_INT][eft_INT] = {eft_INT, (void*)dpnp_erf_c_default}; fmap[DPNPFuncName::DPNP_FN_ERF][eft_LNG][eft_LNG] = {eft_LNG, (void*)dpnp_erf_c_default}; diff --git a/tests/skipped_tests.tbl b/tests/skipped_tests.tbl index a41b881ae7b..707c0b89724 100644 --- a/tests/skipped_tests.tbl +++ b/tests/skipped_tests.tbl @@ -202,8 +202,6 @@ tests/test_random.py::TestPermutationsTestShuffle::test_shuffle1[lambda x: (dpnp tests/test_random.py::TestPermutationsTestShuffle::test_shuffle1[lambda x: dpnp.asarray([(i, i) for i in x], [("a", object), ("b", dpnp.int32)])]] tests/test_random.py::TestPermutationsTestShuffle::test_shuffle1[lambda x: dpnp.asarray(x).astype(dpnp.int8)] -tests/test_sort.py::test_partition[[[1, 0], [3, 0]]-float32-1] - tests/third_party/cupy/core_tests/test_ndarray_complex_ops.py::TestAngle::test_angle tests/third_party/cupy/core_tests/test_ndarray_complex_ops.py::TestRealImag::test_imag tests/third_party/cupy/core_tests/test_ndarray_complex_ops.py::TestRealImag::test_imag_inplace diff --git a/tests/skipped_tests_gpu.tbl b/tests/skipped_tests_gpu.tbl index 36b17d5edbc..1f50dfd5b50 100644 --- a/tests/skipped_tests_gpu.tbl +++ b/tests/skipped_tests_gpu.tbl @@ -349,7 +349,6 @@ tests/test_linalg.py::test_matrix_rank[None-[[1, 2], [3, 4]]-int32] tests/test_random.py::TestPermutationsTestShuffle::test_shuffle1[lambda x: (dpnp.asarray([(i, i) for i in x], [("a", int), ("b", int)]).view(dpnp.recarray))] tests/test_random.py::TestPermutationsTestShuffle::test_shuffle1[lambda x: dpnp.asarray([(i, i) for i in x], [("a", object), ("b", dpnp.int32)])]] tests/test_random.py::TestPermutationsTestShuffle::test_shuffle1[lambda x: dpnp.asarray(x).astype(dpnp.int8)] -tests/test_sort.py::test_partition[[[1, 0], [3, 0]]-float32-1] tests/third_party/cupy/core_tests/test_ndarray_complex_ops.py::TestAngle::test_angle tests/third_party/cupy/core_tests/test_ndarray_complex_ops.py::TestRealImag::test_imag diff --git a/tests/test_sort.py b/tests/test_sort.py index 975c654cbb9..cfcef2c3db0 100644 --- a/tests/test_sort.py +++ b/tests/test_sort.py @@ -30,8 +30,11 @@ def test_partition(array, dtype, kth): a = dpnp.array(array, dtype) p = dpnp.partition(a, kth) - assert (p[0:kth] <= p[kth]).all() - assert (p[kth] <= p[kth + 1:]).all() + # TODO: remove once dpnp.less_equal() support complex types + p = p.asnumpy() + + assert (p[..., 0:kth] <= p[..., kth:kth + 1]).all() + assert (p[..., kth:kth + 1] <= p[..., kth + 1:]).all() @pytest.mark.usefixtures("allow_fall_back_on_numpy")