@@ -495,22 +495,24 @@ cdef utils.dpnp_descriptor call_fptr_2in_1out_strides(DPNPFuncName fptr_name,
495
495
496
496
# get FPTR function and result type
497
497
cdef fptr_2in_1out_strides_t func = NULL
498
+ cdef DPNPFuncType return_type = DPNP_FT_NONE
498
499
if fptr_name != DPNP_FN_DIVIDE_EXT or result_sycl_device.has_aspect_fp64:
499
- result_type = dpnp_DPNPFuncType_to_dtype( < size_t > kernel_data.return_type)
500
+ return_type = kernel_data.return_type
500
501
func = < fptr_2in_1out_strides_t > kernel_data.ptr
501
502
else :
502
- result_type = dpnp_DPNPFuncType_to_dtype( < size_t > kernel_data.return_type_no_fp64)
503
+ return_type = kernel_data.return_type_no_fp64
503
504
func = < fptr_2in_1out_strides_t > kernel_data.ptr_no_fp64
504
505
505
506
if out is None :
506
507
""" Create result array with type given by FPTR data """
507
508
result = utils.create_output_descriptor(result_shape,
508
- kernel_data. return_type,
509
+ return_type,
509
510
None ,
510
511
device = result_sycl_device,
511
512
usm_type = result_usm_type,
512
513
sycl_queue = result_sycl_queue)
513
514
else :
515
+ result_type = dpnp_DPNPFuncType_to_dtype(< size_t > return_type)
514
516
if out.dtype != result_type:
515
517
utils.checker_throw_value_error(func_name, ' out.dtype' , out.dtype, result_type)
516
518
if out.shape != result_shape:
0 commit comments