[BUILD] Stop generating sym kernels by default (#1907)

* Stop generating sym kernels by default

[ROCm/rccl commit: 93d86dd8e3]
Этот коммит содержится в:
Bertan Dogancay authored and GitHub committed 2025-09-15 12:19:35 -04:00
1 родитель 1b8a616247
Коммит 546b37e35a
3 files changed
+30 -12

No files matched your search

+7
Просмотреть файл
@@ -96,10 +96,15 @@ ncclResult_t ncclInitKernelsForDevice(int cudaArch, int maxSharedMem, size_t* ma
CUDACHECK(hipDeviceGetAttribute(&WarpSize, hipDeviceAttributeWarpSize, cudaDev));
int ncclMaxSharedMem = rcclShmemDynamicSize(cudaArch, WarpSize);
#ifdef GENERATE_SYM_KERNELS
for (int sym=0; sym <= 1; sym++) {
int kcount = sym==0 ? KernelCount : ncclSymKernelCount;
for (int k=0; k < kcount; k++) {
void* fn = sym==0 ? ncclKerns[k].kernelFn : ncclSymKernelList[k];
#else
for (int k = 0; k < KernelCount; k++) {
void* fn = ncclKerns[k].kernelFn;
#endif
cudaFuncAttributes attr = {0};
if (fn == nullptr) continue;
@@ -130,7 +135,9 @@ ncclResult_t ncclInitKernelsForDevice(int cudaArch, int maxSharedMem, size_t* ma
}
next_kernel:;
}
#ifdef GENERATE_SYM_KERNELS
}
#endif
return result;
}