...

Author: decarvalhojunior-fh

Project.toml

@@ -3,13 +3,15 @@ uuid = "236308dc-ae00-46af-a4e9-4f62f68ef0a5"
 version = "1.11.0"
 
 [deps]
+Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
 MPIClusterManagers = "e7922434-ae4b-11e9-05c5-9780451d2c66"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 Serialization = "9e88b42a-f829-5b0c-bbe9-9e923198166b"
 Sockets = "6462fe0b-24de-5631-8697-dd941f90decc"
 
 [compat]
 Aqua = "0.8.10"
+Distributed = "1.11.0"
 LinearAlgebra = "1"
 MPIClusterManagers = "0.2.4"
 Random = "1"

src/Multicluster.jl

@@ -23,6 +23,7 @@ import Serialization: serialize, deserialize
 import Sockets: connect, wait_connected
 
 using MPIClusterManagers
+using Distributed
 
 # NOTE: clusterserialize.jl imports additional symbols from Serialization for use
 

src/multicluster_exts.jl

@@ -20,7 +20,13 @@ struct Node
     pid::Integer
     function Node(cid, pid)
         @assert haskey(cluster_table[], cid)
-        @assert(@fetchfrom(cid, in(pid, workers(role=:master))))
+        #@assert(@fetchfrom(cid, in(pid, workers(role=:master))))
+        try
+            b = remotecall_fetch(w -> in(w, Distributed.workers(role=:master)), cid, pid)
+            @assert b
+        catch e
+            @info e
+        end        
         new(cid, pid)
     end
 end
@@ -38,6 +44,7 @@ function addcluster(access_node, nw; kwargs...)
     master_id = addprocs([access_node]; access_node_args...)
 
     @everywhere master_id @eval using MPIClusterManagers
+    @everywhere master_id @eval using Distributed 
 
     @info "master_id = $master_id"
 
@@ -46,15 +53,15 @@ function addcluster(access_node, nw; kwargs...)
     @info "compute_node_args = $compute_node_args"
 
     wpids = if MPI 
-               remotecall_fetch(addprocs, master_id[1], MPIWorkerManager(nw); compute_node_args...)
+               remotecall_fetch(Distributed.addprocs, master_id[1], MPIWorkerManager(nw); compute_node_args...)
             else
                # TODO
                throw("not implemented")
             end
 
-    @info "pids = $wpids"
+    @info "pids ? $wpids"
 
-    cluster_table[][master_id] = ClusterInfo(master_id[1], access_node, nw, access_node_args, compute_node_args, [wpids])
+    cluster_table[][master_id[1]] = ClusterInfo(master_id[1], access_node, nw, access_node_args, compute_node_args, wpids)
 
     return Cluster(master_id[1])
 end