Fix ROCm-aware MPI extension (JuliaParallel#763)

* Fix conversion to Ptr
giordano · Sep 5, 2023 · c1794e9 · c1794e9
1 parent 786ef51
commit c1794e9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "MPI"
 uuid = "da04e1cc-30fd-572f-bb4f-1f8673147195"
 authors = []
-version = "0.20.14"
+version = "0.20.15"
 
 [deps]
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
@@ -19,7 +19,7 @@ Serialization = "9e88b42a-f829-5b0c-bbe9-9e923198166b"
 Sockets = "6462fe0b-24de-5631-8697-dd941f90decc"
 
 [compat]
-AMDGPU = "0.3, 0.4, 0.5"
+AMDGPU = "0.5.7"
 CUDA = "3, 4"
 DocStringExtensions = "0.8, 0.9"
 MPIPreferences = "0.1.8"

diff --git a/ext/AMDGPUExt.jl b/ext/AMDGPUExt.jl
@@ -9,7 +9,7 @@ function Base.cconvert(::Type{MPIPtr}, A::AMDGPU.ROCArray{T}) where T
 end
 
 function Base.unsafe_convert(::Type{MPIPtr}, X::AMDGPU.ROCArray{T}) where T
-    reinterpret(MPIPtr, Base.unsafe_convert(Ptr{T}, X.buf.ptr+X.offset))
+    reinterpret(MPIPtr, Base.unsafe_convert(Ptr{T}, X))
 end
 
 # only need to define this for strided arrays: all others can be handled by generic machinery