From 2ca3f2e1ffa193d8f8b2c6eaca643907f03d69cc Mon Sep 17 00:00:00 2001 From: Lukasz Stafiniak Date: Tue, 17 Dec 2024 18:05:44 +0100 Subject: [PATCH] Comment out a tiny cudajit.0.6.2 dependency --- arrayjit/lib/cuda_backend.cudajit.ml | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/arrayjit/lib/cuda_backend.cudajit.ml b/arrayjit/lib/cuda_backend.cudajit.ml index 085f854b..260a0088 100644 --- a/arrayjit/lib/cuda_backend.cudajit.ml +++ b/arrayjit/lib/cuda_backend.cudajit.ml @@ -189,8 +189,9 @@ module Fresh () = struct let same_device = dev.ordinal = src.stream.device.ordinal in let size_in_bytes = Lazy.force tn.Tn.size_in_bytes in let memcpy ~dst_ptr = - if same_device && Cu.Deviceptr.equal dst_ptr src_ptr then () - else if same_device then + (* FIXME: coming in cudajit.0.6.2. *) + (* if same_device && Cu.Deviceptr.equal dst_ptr src_ptr then () else *) + if same_device then Cu.Stream.memcpy_D_to_D ~size_in_bytes ~dst:dst_ptr ~src:src_ptr dst.stream.runner else Cu.Stream.memcpy_peer ~size_in_bytes ~dst:dst_ptr ~dst_ctx:(ctx_of dst) ~src:src_ptr