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 b4e65b3 commit 969e311Copy full SHA for 969e311
src/cuda.jl
@@ -1,6 +1,7 @@
1
import .CUDA
2
3
function Base.cconvert(::Type{MPIPtr}, buf::CUDA.CuArray{T}) where T
4
+ CUDA.synchronize(CUDA.CuDefaultStream())
5
Base.cconvert(CUDA.CuPtr{T}, buf) # returns DeviceBuffer
6
end
7
test/runtests.jl
@@ -4,6 +4,8 @@ using Test, MPI
using DoubleFloats
if get(ENV,"JULIA_MPI_TEST_ARRAYTYPE","") == "CuArray"
import CUDA
+ CUDA.version()
8
+ CUDA.precompile_runtime()
9
ArrayType = CUDA.CuArray
10
else
11
ArrayType = Array
0 commit comments