Disable Colltrace for --fast option (#778)

* Disable Colltrace for --fast option * Limit nprocs for CI
2023-06-21 16:16:09 -04:00
parent 399d31ed40
commit 0c77c66221
8 changed files with 85 additions and 43 deletions
@@ -224,7 +224,10 @@ static float ncclTopoXGMISpeed(int gcn) {
  return gcn == 910 ? MI200_XGMI_WIDTH : VEGA_XGMI_WIDTH;
 }

-#define ncclGetKernelIndex(p_comm) \
-  ((p_comm)->collTraceThread ? 1 : 0)
+#if ENABLE_COLLTRACE
+  #define ncclGetKernelIndex(p_comm) ((p_comm)->collTraceThread ? 1 : 0)
+#else
+  #define ncclGetKernelIndex(p_comm) (0)
+#endif

 #endif