commit 43adc5ba05546b8eb88efd37231f1314e0a5b2f3
author Aapo Kyrola <akyrola@fb.com> Sat Oct 07 21:50:00 2017 -0700
committer Facebook Github Bot <facebook-github-bot@users.noreply.github.com> Sat Oct 07 22:06:11 2017 -0700
tree 6964455a1900a190a192ac3cfd045efe24d5f044
parent 463bcd00ea906a76973c504dbbaf430f13ddff17

Add nodename to ONE, iteration_mutex etc.

Summary: Similar as with Iter, LR.

Reviewed By: azzolini

Differential Revision: D6005817

fbshipit-source-id: 6d1260791d1acb3df957315eb9156eac183ee25c

caffe2/python/optimizer.py

diff --git a/caffe2/python/optimizer.py b/caffe2/python/optimizer.py
index 418f711..7cca45b 100644
--- a/caffe2/python/optimizer.py
+++ b/caffe2/python/optimizer.py
@@ -110,7 +110,9 @@
                     [], optimization_iter_blob, shape=[1],
                     value=iter_val,
                     dtype=core.DataType.INT64)
-                iter_mutex = param_init_net.CreateMutex([], ["iteration_mutex"])
+                iter_mutex = param_init_net.CreateMutex(
+                    [], ["iteration_mutex" + node_name]
+                )
                 net.AtomicIter([iter_mutex, iteration], [iteration])
         else:
             iteration = param_init_net.GetBlobRef(optimization_iter_blob)
@@ -223,7 +225,7 @@
         # to include device information.
         ONE = param_init_net.ConstantFill(
             [],
-            "ONE_{}_{}".format(dev.device_type, dev.cuda_gpu_id),
+            "ONE_{}_{}{}".format(dev.device_type, dev.cuda_gpu_id, dev.node_name),
             shape=[1],
             value=1.0
         )