Skip to content

Commit 9fb287c

Browse files
densmirnoleksandr-pavlyk
authored andcommitted
Change output array creation in dpnp_diag (part 2)
1 parent a1597d0 commit 9fb287c

File tree

1 file changed

+1
-8
lines changed

1 file changed

+1
-8
lines changed

dpnp/dpnp_algo/dpnp_algo_arraycreation.pyx

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -123,20 +123,13 @@ cpdef utils.dpnp_descriptor dpnp_diag(utils.dpnp_descriptor v, int k):
123123
v_obj = v.get_array()
124124

125125
# TODO need to call dpnp_container.zeros instead
126-
result_obj = dpnp_container.asarray(dpnp.zeros(result_shape, dtype=v.dtype),
127-
dtype=v.dtype,
128-
device=v_obj.sycl_device,
129-
usm_type=v_obj.usm_type,
130-
sycl_queue=v_obj.sycl_queue)
126+
result_obj = dpnp.zeros(result_shape, dtype=v.dtype).to_device(v_obj.sycl_device)
131127
cdef utils.dpnp_descriptor result = dpnp_descriptor(result_obj)
132128

133129
cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(v.dtype)
134130

135131
cdef DPNPFuncData kernel_data = get_dpnp_function_ptr(DPNP_FN_DIAG_EXT, param1_type, param1_type)
136132

137-
result_type = dpnp_DPNPFuncType_to_dtype(< size_t > kernel_data.return_type)
138-
139-
140133
result_sycl_queue = result.get_array().sycl_queue
141134

142135
cdef c_dpctl.SyclQueue q = <c_dpctl.SyclQueue> result_sycl_queue

0 commit comments

Comments
 (0)