diff --git a/numba_dpex/dpnp_iface/arrayobj.py b/numba_dpex/dpnp_iface/arrayobj.py index 90e3f922ad..694ac37db5 100644 --- 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,6 +408,7 @@ 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" @@ -414,7 +416,7 @@ def ol_dpnp_ones( if _ndim: ret_ty = build_dpnp_ndarray( _ndim, - layout=order, + layout=_layout, dtype=_dtype, usm_type=_usm_type, device=_device,