fix(rccl): disable symmetric kernels when GENERATE_SYM_KERNELS is OFF
This commit is contained in:
@@ -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,
|
||||
|
||||
Fai riferimento in un nuovo problema
Block a user