fix

microsoft · Nov 1, 2024 · 899bccf · 899bccf
1 parent 4ce8d87
commit 899bccf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/include/mscclpp/gpu_data_types.hpp b/include/mscclpp/gpu_data_types.hpp
@@ -15,6 +15,7 @@ using __bfloat162 = __hip_bfloat162;
 
 #else
 
+#include <cuda_runtime.h>
 #include <cuda_fp16.h>
 #if (CUDART_VERSION >= 11000)
 #include <cuda_bf16.h>