Skip to content

Commit 54f006f

Browse files
author
Vahid Tavanashad
committed
address comments
1 parent 41f4646 commit 54f006f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dpnp/dpnp_utils/dpnp_utils_reduction.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@
2929
__all__ = ["dpnp_wrap_reduction_call"]
3030

3131

32-
def dpnp_wrap_reduction_call(a, out, _reduction_fn, res_dt, **kwargs):
32+
def dpnp_wrap_reduction_call(usm_a, out, _reduction_fn, res_dt, **kwargs):
3333
"""Wrap a reduction call from dpctl.tensor interface."""
3434

3535
input_out = out
@@ -45,5 +45,5 @@ def dpnp_wrap_reduction_call(a, out, _reduction_fn, res_dt, **kwargs):
4545
usm_out = dpnp.get_usm_ndarray(out)
4646

4747
kwargs["out"] = usm_out
48-
res_usm = _reduction_fn(a, **kwargs)
48+
res_usm = _reduction_fn(usm_a, **kwargs)
4949
return dpnp.get_result_array(res_usm, input_out, casting="unsafe")

0 commit comments

Comments
 (0)