Fix NPKit for SendRecv (#1651)

[ROCm/rccl commit: ac8ec4c08c]
This commit is contained in:
Bertan Dogancay
2025-04-21 12:34:47 -04:00
committed by GitHub
orang tua 58ee618194
melakukan aac829125c
2 mengubah file dengan 3 tambahan dan 2 penghapusan
+1
Melihat File
@@ -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 {