Julia中使用ccall及MPI绑定时如何防止缓冲区被垃圾回收？

问题现象 #

自定义MPI Recv 函数时，使用 GC.@preserve 包裹 @ccall 并直接传入 buf::Ref{T} 并指定类型为 Ptr{T}，会导致缓冲区被垃圾回收，无法正确接收数据；但只要在 GC.@preserve 块内显式调用 Base.unsafe_convert(Ptr{T}, buf) 或者访问 buf[]，就能正常保护缓冲区，数据接收正常。

能正常工作的实现：

function MPI_Rec(buf::Ref{T}, count::C_int, datatype::C_MPI_Datatype, source::C_int, tag::C_int, comm::C_MPI_Comm, status::Ptr{C_MPI_Status})::C_int where T
    GC.@preserve buf begin
        Base.unsafe_convert(Ptr{T}, buf)
        return @ccall "libmpi".MPI_Recv(buf::Ptr{T}, count::C_int, datatype::C_MPI_Datatype, source::C_int, tag::C_int, comm::C_MPI_Comm, status::Ptr{C_MPI_Status})::C_int
    end
end

失效的实现：

function MPI_Rec(buf::Ref{T}, count::C_int, datatype::C_MPI_Datatype, source::C_int, tag::C_int, comm::C_MPI_Comm, status::Ptr{C_MPI_Status})::C_int where T
    GC.@preserve buf begin
        return @ccall "libmpi".MPI_Recv(buf::Ptr{T}, count::C_int, datatype::C_MPI_Datatype, source::C_int, tag::C_int, comm::C_MPI_Comm, status::Ptr{C_MPI_Status})::C_int
    end
end

原因分析 #

Julia的 GC.@preserve 通过标记变量在块内不会被GC回收来工作，但 @ccall 宏中的类型标注（buf::Ptr{T}）是在宏展开阶段处理的，这个过程并未让编译器将 buf 识别为“正在被使用”的对象——也就是说，GC.@preserve 块内如果没有直接对 buf 的显式引用操作，编译器可能认为 buf 已不再需要，从而允许GC在 @ccall 执行期间回收它。

而显式调用 Base.unsafe_convert(Ptr{T}, buf) 或者访问 buf[]，会生成对 buf 的直接引用，让编译器明确知道该对象在 GC.@preserve 块内仍在被使用，确保缓冲区在MPI接收操作完成前不会被回收。

另外，将 buf 声明为全局变量时能正常工作，是因为全局变量生命周期更长，GC不会轻易对其进行回收，避开了局部变量的GC时机问题。

正确实现方式 #

最可靠的做法是显式获取缓冲区的指针，再传入 @ccall，确保 buf 被正确引用和保护：

function Recv(buf::Ref{T}, count::C_int, datatype::C_MPI_Datatype, source::C_int, tag::C_int, comm::C_MPI_Comm, status::Ptr{C_MPI_Status})::C_int where T
    GC.@preserve buf begin
        ptr = Base.unsafe_convert(Ptr{T}, buf)
        return @ccall "libmpi".MPI_Recv(ptr::Ptr{T}, count::C_int, datatype::C_MPI_Datatype, source::C_int, tag::C_int, comm::C_MPI_Comm, status::Ptr{C_MPI_Status})::C_int
    end 
end

这种写法不仅能确保 GC.@preserve 生效，也让代码逻辑更清晰，明确展示指针转换过程，避免隐式转换带来的GC问题。

验证说明 #

调用上述正确实现的 Recv 函数时，局部声明的 Ref 对象（如示例中的 val1 = Ref{Cdouble}(0.1)）会在 GC.@preserve 块内被持续引用，直到MPI接收操作完成，不会被GC提前回收，从而能正确接收并存储数据。

内容的提问来源于stack exchange，提问作者jorge ezpeleta