Merge pull request #987 from IntelPython/dpnp_overload_fix

Fixed layout in dpnp overload.
IntelPython · Mar 31, 2023 · 3478aba · 3478aba
2 parents de2a895 + 125585a
commit 3478aba
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/numba_dpex/dpnp_iface/arrayobj.py b/numba_dpex/dpnp_iface/arrayobj.py
@@ -326,6 +326,7 @@ def ol_dpnp_zeros(
     """
 
     _ndim = _ty_parse_shape(shape)
+    _layout = _parse_layout(order)
     _dtype = _parse_dtype(dtype)
     _usm_type = _parse_usm_type(usm_type) if usm_type is not None else "device"
     _device = (
@@ -334,7 +335,7 @@ def ol_dpnp_zeros(
     if _ndim:
         ret_ty = build_dpnp_ndarray(
             _ndim,
-            layout=order,
+            layout=_layout,
             dtype=_dtype,
             usm_type=_usm_type,
             device=_device,
@@ -407,14 +408,15 @@ def ol_dpnp_ones(
 
     _ndim = _ty_parse_shape(shape)
     _dtype = _parse_dtype(dtype)
+    _layout = _parse_layout(order)
     _usm_type = _parse_usm_type(usm_type) if usm_type is not None else "device"
     _device = (
         _parse_device_filter_string(device) if device is not None else "unknown"
     )
     if _ndim:
         ret_ty = build_dpnp_ndarray(
             _ndim,
-            layout=order,
+            layout=_layout,
             dtype=_dtype,
             usm_type=_usm_type,
             device=_device,