Example code fix for pytorch/tutorials#1619

bowangbj · bowangbj · commit 5c47bb1b550e · 2021-07-27T14:52:04.000-07:00
Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:

[ghstack-poisoned]
diff --git a/distributed/rpc/ddp_rpc/main.py b/distributed/rpc/ddp_rpc/main.py
@@ -137,7 +137,7 @@ def run_worker(rank, world_size):
         for trainer_rank in [0, 1]:
             trainer_name = "trainer{}".format(trainer_rank)
             fut = rpc.rpc_async(
-                trainer_name, _run_trainer, args=(remote_emb_module, rank)
+                trainer_name, _run_trainer, args=(remote_emb_module, trainer_rank)
             )
             futs.append(fut)
 

Original file line number	Diff line number	Diff line change
`@@ -137,7 +137,7 @@ def run_worker(rank, world_size):`
`137`	`137`	`for trainer_rank in [0, 1]:`
`138`	`138`	`trainer_name = "trainer{}".format(trainer_rank)`
`139`	`139`	`fut = rpc.rpc_async(`
`140`		`- trainer_name, _run_trainer, args=(remote_emb_module, rank)`
	`140`	`+ trainer_name, _run_trainer, args=(remote_emb_module, trainer_rank)`
`141`	`141`	`)`
`142`	`142`	`futs.append(fut)`
`143`	`143`