Default to minimal 2 rings and improve LL loop

[ROCm/rccl commit: 4b77a16f3f]
2019-08-14 14:12:56 -07:00
parent 1feef99e7d
commit 93c44e96cb
2 changed files with 5 additions and 3 deletions
@@ -171,7 +171,9 @@ static ncclResult_t fillCoords(int nranks, int* matrix, int* coords, int* rankTo
  }
 }

-#ifdef __PPC__
+#if defined(__HIP_PLATFORM_HCC__) || defined(__HCC__)
+#define DEFAULT_MIN_NRINGS 2
+#elif defined(__PPC__)
 // Make the default NCCL_MIN_NRINGS=4 for IBM/Power nodes
 #define DEFAULT_MIN_NRINGS 4
 #else