fix(rccl): disable symmetric kernels when GENERATE_SYM_KERNELS is OFF

Этот коммит содержится в:
Donato Capitella
2026-02-01 08:44:52 +00:00
родитель 8126402d12
Коммит d2ea5d5d4c
+4
Просмотреть файл
@@ -260,6 +260,7 @@ static bool ncclSymkImplemented(ncclFunc_t coll, int/*ncclDevRedOp_t*/ red, nccl
}
static uint32_t ncclSymkMask(struct ncclComm* comm, ncclFunc_t coll, int/*ncclDevRedOp_t*/ red, ncclDataType_t ty, size_t nElts) {
#if defined(GENERATE_SYM_KERNELS) && GENERATE_SYM_KERNELS
uint32_t kmask = kernelMask_coll(coll);
kmask &= kernelMask_user();
@@ -299,6 +300,9 @@ static uint32_t ncclSymkMask(struct ncclComm* comm, ncclFunc_t coll, int/*ncclDe
if (nBusBytes >= 32*(size_t(2)<<30)) kmask = 0;
return kmask;
#else
return 0;
#endif
}
bool ncclSymkAvailable(struct ncclComm* comm, ncclFunc_t coll, int/*ncclDevRedOp_t*/ red,