Skip to content

Commit e59c1f4

Browse files
vtavanavtavana
authored
update cupy math tests (#1589)
* update cupy math tests * address comments * remove test_sumprod.py for public CI
1 parent c951d41 commit e59c1f4

File tree

5 files changed

+54
-269
lines changed

5 files changed

+54
-269
lines changed

‎.github/workflows/conda-package.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,13 @@ env:
3636
third_party/cupy/manipulation_tests/test_join.py
3737
third_party/cupy/manipulation_tests/test_rearrange.py
3838
third_party/cupy/manipulation_tests/test_transpose.py
39+
third_party/cupy/math_tests/test_arithmetic.py
3940
third_party/cupy/math_tests/test_explog.py
41+
third_party/cupy/math_tests/test_floating.py
42+
third_party/cupy/math_tests/test_hyperbolic.py
43+
third_party/cupy/math_tests/test_matmul.py
4044
third_party/cupy/math_tests/test_misc.py
45+
third_party/cupy/math_tests/test_rounding.py
4146
third_party/cupy/math_tests/test_trigonometric.py
4247
third_party/cupy/sorting_tests/test_sort.py
4348
VER_JSON_NAME: 'version.json'

‎dpnp/backend/kernels/dpnp_krnl_elemwise.cpp

Lines changed: 42 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -409,9 +409,21 @@ static void func_map_init_elemwise_1arg_2type(func_map_t &fmap)
409409
eft_DBL, (void *)dpnp_degrees_c_default<double, double>};
410410

411411
fmap[DPNPFuncName::DPNP_FN_DEGREES_EXT][eft_INT][eft_INT] = {
412-
eft_DBL, (void *)dpnp_degrees_c_ext<int32_t, double>};
412+
get_default_floating_type(),
413+
(void *)dpnp_degrees_c_ext<
414+
int32_t, func_type_map_t::find_type<get_default_floating_type()>>,
415+
get_default_floating_type<std::false_type>(),
416+
(void *)dpnp_degrees_c_ext<
417+
int32_t, func_type_map_t::find_type<
418+
get_default_floating_type<std::false_type>()>>};
413419
fmap[DPNPFuncName::DPNP_FN_DEGREES_EXT][eft_LNG][eft_LNG] = {
414-
eft_DBL, (void *)dpnp_degrees_c_ext<int64_t, double>};
420+
get_default_floating_type(),
421+
(void *)dpnp_degrees_c_ext<
422+
int64_t, func_type_map_t::find_type<get_default_floating_type()>>,
423+
get_default_floating_type<std::false_type>(),
424+
(void *)dpnp_degrees_c_ext<
425+
int64_t, func_type_map_t::find_type<
426+
get_default_floating_type<std::false_type>()>>};
415427
fmap[DPNPFuncName::DPNP_FN_DEGREES_EXT][eft_FLT][eft_FLT] = {
416428
eft_FLT, (void *)dpnp_degrees_c_ext<float, float>};
417429
fmap[DPNPFuncName::DPNP_FN_DEGREES_EXT][eft_DBL][eft_DBL] = {
@@ -427,9 +439,21 @@ static void func_map_init_elemwise_1arg_2type(func_map_t &fmap)
427439
eft_DBL, (void *)dpnp_exp2_c_default<double, double>};
428440

429441
fmap[DPNPFuncName::DPNP_FN_EXP2_EXT][eft_INT][eft_INT] = {
430-
eft_DBL, (void *)dpnp_exp2_c_ext<int32_t, double>};
442+
get_default_floating_type(),
443+
(void *)dpnp_exp2_c_ext<
444+
int32_t, func_type_map_t::find_type<get_default_floating_type()>>,
445+
get_default_floating_type<std::false_type>(),
446+
(void *)dpnp_exp2_c_ext<
447+
int32_t, func_type_map_t::find_type<
448+
get_default_floating_type<std::false_type>()>>};
431449
fmap[DPNPFuncName::DPNP_FN_EXP2_EXT][eft_LNG][eft_LNG] = {
432-
eft_DBL, (void *)dpnp_exp2_c_ext<int64_t, double>};
450+
get_default_floating_type(),
451+
(void *)dpnp_exp2_c_ext<
452+
int64_t, func_type_map_t::find_type<get_default_floating_type()>>,
453+
get_default_floating_type<std::false_type>(),
454+
(void *)dpnp_exp2_c_ext<
455+
int64_t, func_type_map_t::find_type<
456+
get_default_floating_type<std::false_type>()>>};
433457
fmap[DPNPFuncName::DPNP_FN_EXP2_EXT][eft_FLT][eft_FLT] = {
434458
eft_FLT, (void *)dpnp_exp2_c_ext<float, float>};
435459
fmap[DPNPFuncName::DPNP_FN_EXP2_EXT][eft_DBL][eft_DBL] = {
@@ -526,9 +550,21 @@ static void func_map_init_elemwise_1arg_2type(func_map_t &fmap)
526550
eft_DBL, (void *)dpnp_radians_c_default<double, double>};
527551

528552
fmap[DPNPFuncName::DPNP_FN_RADIANS_EXT][eft_INT][eft_INT] = {
529-
eft_DBL, (void *)dpnp_radians_c_ext<int32_t, double>};
553+
get_default_floating_type(),
554+
(void *)dpnp_radians_c_ext<
555+
int32_t, func_type_map_t::find_type<get_default_floating_type()>>,
556+
get_default_floating_type<std::false_type>(),
557+
(void *)dpnp_radians_c_ext<
558+
int32_t, func_type_map_t::find_type<
559+
get_default_floating_type<std::false_type>()>>};
530560
fmap[DPNPFuncName::DPNP_FN_RADIANS_EXT][eft_LNG][eft_LNG] = {
531-
eft_DBL, (void *)dpnp_radians_c_ext<int64_t, double>};
561+
get_default_floating_type(),
562+
(void *)dpnp_radians_c_ext<
563+
int64_t, func_type_map_t::find_type<get_default_floating_type()>>,
564+
get_default_floating_type<std::false_type>(),
565+
(void *)dpnp_radians_c_ext<
566+
int64_t, func_type_map_t::find_type<
567+
get_default_floating_type<std::false_type>()>>};
532568
fmap[DPNPFuncName::DPNP_FN_RADIANS_EXT][eft_FLT][eft_FLT] = {
533569
eft_FLT, (void *)dpnp_radians_c_ext<float, float>};
534570
fmap[DPNPFuncName::DPNP_FN_RADIANS_EXT][eft_DBL][eft_DBL] = {

0 commit comments

Comments
 (0)