diff --git a/src/executor/execution_kernel.cu b/src/executor/execution_kernel.cu index f5a24ff0f..d5e07a3da 100644 --- a/src/executor/execution_kernel.cu +++ b/src/executor/execution_kernel.cu @@ -3,6 +3,7 @@ #include "execution_kernel.hpp" +#if defined(MSCCLPP_DEVICE_CUDA) namespace mscclpp { void ExecutionKernel::launchKernel(int rank, int nthreadblocks, int nthreads, void* src, void* dst, void* scratch, DataType dataType, DeviceExecutionPlan* plan, size_t sharedMemSize, @@ -27,3 +28,4 @@ void ExecutionKernel::launchKernel(int rank, int nthreadblocks, int nthreads, vo } } } // namespace mscclpp +#endif