Skip to content

Commit 1a5c09c

Browse files
authored
Fix DPNP/DPCTL F-contiguous data processing (#1825)
* Fix DPNP/DPCTL F-contiguous data processing * Code formatting * Fix linter error
1 parent 8bf52c2 commit 1a5c09c

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

sklearnex/_device_offload.py

+8-1
Original file line numberDiff line numberDiff line change
@@ -127,8 +127,15 @@ def _transfer_to_host(queue, *data):
127127
queue = usm_iface["syclobj"]
128128

129129
buffer = as_usm_memory(item).copy_to_host()
130+
order = "C"
131+
if usm_iface["strides"] is not None:
132+
if usm_iface["strides"][0] < usm_iface["strides"][1]:
133+
order = "F"
130134
item = np.ndarray(
131-
shape=usm_iface["shape"], dtype=usm_iface["typestr"], buffer=buffer
135+
shape=usm_iface["shape"],
136+
dtype=usm_iface["typestr"],
137+
buffer=buffer,
138+
order=order,
132139
)
133140
has_usm_data = True
134141
else:

0 commit comments

Comments
 (0)