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 19dfd0d commit 2ca3f2eCopy full SHA for 2ca3f2e
arrayjit/lib/cuda_backend.cudajit.ml
@@ -189,8 +189,9 @@ module Fresh () = struct
189
let same_device = dev.ordinal = src.stream.device.ordinal in
190
let size_in_bytes = Lazy.force tn.Tn.size_in_bytes in
191
let memcpy ~dst_ptr =
192
- if same_device && Cu.Deviceptr.equal dst_ptr src_ptr then ()
193
- else if same_device then
+ (* FIXME: coming in cudajit.0.6.2. *)
+ (* if same_device && Cu.Deviceptr.equal dst_ptr src_ptr then () else *)
194
+ if same_device then
195
Cu.Stream.memcpy_D_to_D ~size_in_bytes ~dst:dst_ptr ~src:src_ptr dst.stream.runner
196
else
197
Cu.Stream.memcpy_peer ~size_in_bytes ~dst:dst_ptr ~dst_ctx:(ctx_of dst) ~src:src_ptr
0 commit comments