From 899bccff85b0b9da3d91acab309fec0952260a9f Mon Sep 17 00:00:00 2001
From: Binyang Li <binyli@microsoft.com>
Date: Fri, 1 Nov 2024 11:42:16 +0000
Subject: [PATCH] fix

---
 include/mscclpp/gpu_data_types.hpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/include/mscclpp/gpu_data_types.hpp b/include/mscclpp/gpu_data_types.hpp
index f947ea7f4..cb6c96107 100644
--- 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>