Fix MSCCL work FIFO out-of-bound issue (#935)

Цей коміт міститься в:
Ziyue Yang
2023-10-28 02:24:52 +08:00
зафіксовано GitHub
джерело 96ec3ffe2e
коміт 4c117e5335
3 змінених файлів з 13 додано та 13 видалено
+2 -2
Переглянути файл
@@ -15,7 +15,7 @@
#define MSCCL_MAX_NUM_STEPS 64
#define MSCCL_MAX_NUM_THREAD_BLOCKS_PER_CHANNEL 32
#define MSCCL_MAX_NUM_THREAD_BLOCKS (MSCCL_MAX_NUM_THREAD_BLOCKS_PER_CHANNEL * MAXCHANNELS)
#define MSCCL_MAX_NUM_THREAD_BLOCKS 64
#define MSCCL_MAX_COUNT 72 // max concurrent number of msccl chunk transmission
#define MSCCL_MAX_REDUCE_FUSION 16
#define MSCCL_MAX_NUM_ALGOS 1024
@@ -216,7 +216,7 @@ struct mscclStatus {
struct mscclWork* workFifo;
uint32_t* workFifoDone;
uint32_t workFifoSent;
uint32_t workFifoSentPerChannel[MAXCHANNELS];
uint32_t workFifoSentPerThreadBlock[MSCCL_MAX_NUM_THREAD_BLOCKS];
uint32_t workFifoAckdMin;
};