diff --git a/groundingdino/models/GroundingDINO/csrc/MsDeformAttn/ms_deform_attn_cuda.cu b/groundingdino/models/GroundingDINO/csrc/MsDeformAttn/ms_deform_attn_cuda.cu
index d04fae8a..6b878465 100644
--- a/groundingdino/models/GroundingDINO/csrc/MsDeformAttn/ms_deform_attn_cuda.cu
+++ b/groundingdino/models/GroundingDINO/csrc/MsDeformAttn/ms_deform_attn_cuda.cu
@@ -8,6 +8,7 @@
 **************************************************************************************************
 */
 
+#include <cuda_fp16.h>
 #include <vector>
 #include "ms_deform_im2col_cuda.cuh"
 
@@ -153,4 +154,4 @@ std::vector<at::Tensor> ms_deform_attn_cuda_backward(
     };
 }
 
-} // namespace groundingdino
\ No newline at end of file
+} // namespace groundingdino