set MAXCHANNELS to 128

src/include/comm.h

@@ -173,7 +173,7 @@ struct ncclNvlsMcHandleList {
 };
 
 struct channelMasks {
-        uint64_t masks[4];
+        uint64_t masks[MAXCHANNELS/64];
 };
 
 struct ncclKernelPlan {

src/include/device.h

@@ -61,7 +61,8 @@ union ncclLLFifoLine {
 };
 
 #define WARP_SIZE warpSize
-#define MAXCHANNELS 256
+
+#define MAXCHANNELS 128
 #define CHANNEL_LIMIT 16
 
 #define NCCL_MAX_NTHREADS 256