minor update

microsoft · Apr 5, 2024 · 867101e · 867101e
1 parent d38c9ed
commit 867101e
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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