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 8bf52c2 commit 1a5c09cCopy full SHA for 1a5c09c
sklearnex/_device_offload.py
@@ -127,8 +127,15 @@ def _transfer_to_host(queue, *data):
127
queue = usm_iface["syclobj"]
128
129
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"
134
item = np.ndarray(
- 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,
139
)
140
has_usm_data = True
141
else:
0 commit comments