Fix NPKit for SendRecv (#1651)

[ROCm/rccl commit: ac8ec4c08c]
Esse commit está contido em:
Bertan Dogancay
2025-04-21 12:34:47 -04:00
commit de GitHub
commit aac829125c
2 arquivos alterados com 3 adições e 2 exclusões
+1
Ver Arquivo
@@ -383,6 +383,7 @@ constexpr size_t ncclDevWorkSize(enum ncclDevWorkType type) {
#define NCCL_MAX_DEV_WORK_BATCH_BYTES 128
#define NCCL_MAX_DEV_WORK_BATCH_COLLS (NCCL_MAX_DEV_WORK_BATCH_BYTES/sizeof(ncclDevWorkColl))
#define NCCL_MAX_DEV_WORK_P2P_PER_BATCH 1
#define NCCL_MAX_DEV_WORK_P2P_ELEMENTS 2
struct alignas(16) ncclDevWorkBatch {
union {
struct {