Skip to content

Commit 51369c3

Browse files
Merge pull request #1166 from IntelPython/debug_take_for_main
Corrected logic for kernel function pointer lookup for dpnp_take_c
2 parents 4f9ed25 + 8ddaeb7 commit 51369c3

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

dpnp/dpnp_algo/dpnp_algo_indexing.pyx

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -524,8 +524,9 @@ cpdef utils.dpnp_descriptor dpnp_select(list condlist, list choicelist, default)
524524

525525
cpdef utils.dpnp_descriptor dpnp_take(utils.dpnp_descriptor x1, utils.dpnp_descriptor indices):
526526
cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(x1.dtype)
527+
cdef DPNPFuncType param2_type = dpnp_dtype_to_DPNPFuncType(indices.dtype)
527528

528-
cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_TAKE_EXT, param1_type, param1_type)
529+
cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_TAKE_EXT, param1_type, param2_type)
529530

530531
x1_obj = x1.get_array()
531532

0 commit comments

Comments
 (0)