Broadcasting local arrays among all nodes #3349
-
I'm trying to broadcast an array among all the nodes. I found this piece of code which is almost what I'm looking for: indices{hvd.rank()} = []
# Fill it up
for i in range(hvd.size()):
hvd.broadcast_parameters(indices{i}, root_rank = i) What is a more efficient way of doing this? And does my proposed method work? |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment
-
I think |
Beta Was this translation helpful? Give feedback.
I think
allgather
is the operation you are looking for. :)