[GitHub] ZiyueHuang commented on a change in pull request #8867: fix race when temp space is used in copy & fix instance overwrite in g2c

[GitBox <gi...@apache.org>]

ZiyueHuang commented on a change in pull request #8867: fix race when temp space is used in copy & fix instance overwrite in g2c
URL: https://github.com/apache/incubator-mxnet/pull/8867#discussion_r153779583
 
 

 ##########
 File path: example/model-parallel/matrix_factorization/train.py
 ##########
 @@ -76,7 +76,7 @@
 
     # construct the module
     # map the ctx_group attribute to the context assignment
-    group2ctxs={'dev1':mx.cpu(), 'dev2':[mx.gpu(i) for i in range(num_gpus)]}
+    group2ctxs={'dev1':[mx.cpu()]*num_gpus, 'dev2':[mx.gpu(i) for i in range(num_gpus)]}
 
 Review comment:
   This change is just for better understandability.

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services