From b33ab1ab13cb9e370f5ae1ed059212aad47410c0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mos=C3=A8=20Giordano?= <m.giordano@ucl.ac.uk>
Date: Fri, 17 Apr 2026 10:41:51 +0100
Subject: [PATCH] Fix definition of `Buffer` methods in GPU pkgextensions

---
 Project.toml     | 2 +-
 ext/AMDGPUExt.jl | 4 ++--
 ext/CUDAExt.jl   | 4 ++--
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/Project.toml b/Project.toml
index 5e02e2efe..d0f0f0f56 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,7 +1,7 @@
 name = "MPI"
 uuid = "da04e1cc-30fd-572f-bb4f-1f8673147195"
 authors = []
-version = "0.20.25"
+version = "0.20.26"
 
 [deps]
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
diff --git a/ext/AMDGPUExt.jl b/ext/AMDGPUExt.jl
index 60f68c015..8dc69b80a 100644
--- a/ext/AMDGPUExt.jl
+++ b/ext/AMDGPUExt.jl
@@ -1,7 +1,7 @@
 module AMDGPUExt
 
 using AMDGPU: AMDGPU
-using MPI: MPIPtr, Buffer, Datatype
+using MPI: MPI, MPIPtr, Buffer, Datatype
 
 function Base.cconvert(::Type{MPIPtr}, A::AMDGPU.ROCArray{T}) where T
     A
@@ -19,7 +19,7 @@ function Base.unsafe_convert(::Type{MPIPtr}, V::SubArray{T,N,P,I,true}) where {T
     return reinterpret(MPIPtr, pV)
 end
 
-function Buffer(arr::AMDGPU.ROCArray)
+function MPI.Buffer(arr::AMDGPU.ROCArray)
     Buffer(arr, Cint(length(arr)), Datatype(eltype(arr)))
 end
 
diff --git a/ext/CUDAExt.jl b/ext/CUDAExt.jl
index 51435ea98..b4d60ec97 100644
--- a/ext/CUDAExt.jl
+++ b/ext/CUDAExt.jl
@@ -1,7 +1,7 @@
 module CUDAExt
 
 using CUDA: CUDA
-using MPI: MPIPtr, Buffer, Datatype
+using MPI: MPI, MPIPtr, Buffer, Datatype
 
 function Base.cconvert(::Type{MPIPtr}, buf::CUDA.CuArray{T}) where T
     Base.cconvert(CUDA.CuPtr{T}, buf) # returns DeviceBuffer
@@ -19,7 +19,7 @@ function Base.unsafe_convert(::Type{MPIPtr}, V::SubArray{T,N,P,I,true}) where {T
     return reinterpret(MPIPtr, pV)
 end
 
-function Buffer(arr::CUDA.CuArray)
+function MPI.Buffer(arr::CUDA.CuArray)
     Buffer(arr, Cint(length(arr)), Datatype(eltype(arr)))
 end