We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 497fd42 commit f9f1a41Copy full SHA for f9f1a41
src/cuda.jl
@@ -1,7 +1,7 @@
1
using .CUDA
2
3
function set_device!()
4
- lcomm = MPI.Comm_split_type(COMM_WORLD, MPI.MPI_COMM_TYPE_SHARED, MPI.Comm_rank(COMM_WORLD))
+ lcomm = MPI.Comm_split_type(COMM_WORLD, MPI.COMM_TYPE_SHARED, MPI.Comm_rank(COMM_WORLD))
5
CUDA.device!(MPI.Comm_rank(lcomm) % length(devices()))
6
end
7
0 commit comments