Disable MSCCL kernels at compile time (#834)

* Disable MSCCL kernels at compile time

[ROCm/rccl commit: 64c32d1c5b]
Этот коммит содержится в:
Bertan Dogancay
2023-08-02 09:45:18 -06:00
коммит произвёл GitHub
родитель 677f58edf6
Коммит 74dd9c4807
4 изменённых файлов: 54 добавлений и 27 удалений
+4
Просмотреть файл
@@ -30,7 +30,11 @@ static bool mscclSchedulerTriedLoadAlgo = false;
static std::mutex mscclLifecycleMutex;
bool mscclEnabled() {
#ifdef COMPILE_MSCCL_KERNEL
return rcclParamMscclEnabled();
#else
return false;
#endif
}
void mscclSetIsCallerFlag() {
+4
Просмотреть файл
@@ -9,7 +9,9 @@
#include "transport.h"
#include "msccl/msccl_lifecycle.h"
#ifdef COMPILE_MSCCL_KERNEL
#include "msccl/msccl_kernel.h"
#endif
#include "msccl/msccl_setup.h"
#include "msccl/msccl_status.h"
@@ -243,7 +245,9 @@ static ncclResult_t hostToDevRedOp(
// Except for ncclDevPreMulSum and ncclDevSumPostDiv required by ncclAvg
void* mscclKernelEntries[(ncclNumDevRedOps - 2) * ncclNumTypes * NCCL_NUM_PROTOCOLS] = {
#ifdef COMPILE_MSCCL_KERNEL
MSCCL_KERNEL_ENTRY()
#endif
};
ncclResult_t mscclSetupKernel(const void* sendBuff, void* recvBuff, size_t count,