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 290230a commit f6db04eCopy full SHA for f6db04e
dpctl/tensor/_copy_utils.py
@@ -81,8 +81,15 @@ def _copy_from_numpy_into(dst, np_ary):
81
if not isinstance(dst, dpt.usm_ndarray):
82
raise TypeError("Expected usm_ndarray, got {}".format(type(dst)))
83
src_ary = np.broadcast_to(np_ary, dst.shape)
84
+ copy_q = dst.sycl_queue
85
+ if copy_q.sycl_device.has_aspect_fp64 is False:
86
+ src_ary_dt_c = src_ary.dtype.char
87
+ if src_ary_dt_c == "d":
88
+ src_ary = src_ary.astype(np.float32)
89
+ elif src_ary_dt_c == "D":
90
+ src_ary = src_ary.astype(np.complex64)
91
ti._copy_numpy_ndarray_into_usm_ndarray(
- src=src_ary, dst=dst, sycl_queue=dst.sycl_queue
92
+ src=src_ary, dst=dst, sycl_queue=copy_q
93
)
94
95
0 commit comments