From a5f26eeef7aecc106722fa6a2b37a5f2836a88d7 Mon Sep 17 00:00:00 2001
From: Elton Chen-Yu Ho <chenyu.ho@kaust.edu.sa>
Date: Sun, 22 Aug 2021 17:45:54 +0800
Subject: [PATCH] Set GPU index based on local_rank instead of rank

---
 TensorFlow/Recommendation/NCF/ncf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/TensorFlow/Recommendation/NCF/ncf.py b/TensorFlow/Recommendation/NCF/ncf.py
index 109d6ac8..c68de375 100644
--- a/TensorFlow/Recommendation/NCF/ncf.py
+++ b/TensorFlow/Recommendation/NCF/ncf.py
@@ -227,7 +227,7 @@ def main():
     # Create and run Data Generator in a separate thread
     data_generator = DataGenerator(
         args.seed,
-        hvd.rank(),
+        hvd.local_rank(),
         nb_users,
         nb_items,
         neg_mat,