add 256 channels support

[ROCm/rccl commit: 506f16c506]
Tento commit je obsažen v:
Nusrat Islam
2024-05-15 16:58:28 -05:00
rodič 1934d0f377
revize b0b2aa1166
11 změnil soubory, kde provedl 95 přidání a 53 odebrání
+3 -2
Zobrazit soubor
@@ -61,7 +61,8 @@ union ncclLLFifoLine {
};
#define WARP_SIZE warpSize
#define MAXCHANNELS 64
#define MAXCHANNELS 256
#define NCCL_MAX_NTHREADS 256
#define NCCL_SIMPLE_MAX_NTHREADS NCCL_MAX_NTHREADS
#define NCCL_LL_MAX_NTHREADS NCCL_MAX_NTHREADS
@@ -586,4 +587,4 @@ inline int ncclDevFuncId(int coll, int devRedOp, int type, int algo, int proto)
inline int ncclDevFuncId_P2p() { return ncclDevFuncRowToId[FUNC_INDEX_TOTAL - NCCL_NUM_ONERANK - 1]; }
#endif
#endif