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 ffd16f4 commit 339772eCopy full SHA for 339772e
src/util.jl
@@ -417,6 +417,10 @@ init_GPU_workers(n=nothing) = init_GPU_workers(Val(PARALLEL_WORKER_TYPE), n)
417
418
function init_GPU_workers(::Val{:MPI}, n=nothing; stdout_to_master=false, stderr_to_master=false)
419
420
+ if !CuArrays.functional()
421
+ return
422
+ end
423
+
424
!MPI.Initialized() && MPI.Init()
425
size = MPI.Comm_size(MPI.COMM_WORLD)
426
rank = MPI.Comm_rank(MPI.COMM_WORLD)
@@ -436,6 +440,10 @@ init_GPU_workers(n=nothing) = init_GPU_workers(Val(PARALLEL_WORKER_TYPE), n)
436
440
437
441
function init_GPU_workers(::Val{:procs}, n=nothing)
438
442
443
444
445
446
439
447
if n == nothing
448
n = length(devices())
449
end
0 commit comments