Skip to content

Commit a1597d0

Browse files
densmirnoleksandr-pavlyk
authored andcommitted
Change output array creation in dpnp_diag
1 parent e501f1d commit a1597d0

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

dpnp/dpnp_algo/dpnp_algo_arraycreation.pyx

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -121,12 +121,13 @@ cpdef utils.dpnp_descriptor dpnp_diag(utils.dpnp_descriptor v, int k):
121121
result_shape = (n, )
122122

123123
v_obj = v.get_array()
124-
125-
result_obj = dpnp_container.empty(result_shape,
126-
dtype=v.dtype,
127-
device=v_obj.sycl_device,
128-
usm_type=v_obj.usm_type,
129-
sycl_queue=v_obj.sycl_queue)
124+
125+
# 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)
130131
cdef utils.dpnp_descriptor result = dpnp_descriptor(result_obj)
131132

132133
cdef DPNPFuncType param1_type = dpnp_dtype_to_DPNPFuncType(v.dtype)

0 commit comments

Comments
 (0)