We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ff9b5eb commit ca2c6aaCopy full SHA for ca2c6aa
dpctl/tensor/_reduction.py
@@ -114,15 +114,12 @@ def _reduction_over_axis(
114
res_shape = res_shape + (1,) * red_nd
115
inv_perm = sorted(range(nd), key=lambda d: perm[d])
116
res_shape = tuple(res_shape[i] for i in inv_perm)
117
- return dpt.astype(
118
- dpt.full(
119
- res_shape,
120
- _identity,
121
- dtype=dtype,
122
- usm_type=res_usm_type,
123
- sycl_queue=q,
124
- ),
125
- res_dt,
+ return dpt.full(
+ res_shape,
+ _identity,
+ dtype=res_dt,
+ usm_type=res_usm_type,
+ sycl_queue=q,
126
)
127
if red_nd == 0:
128
return dpt.astype(x, res_dt, copy=False)
0 commit comments