diff --git a/src/core/cuda/sageattention/attn_utils.cuh b/src/core/cuda/sageattention/attn_utils.cuh
index 32f4099a..f112ef26 100644
--- a/src/core/cuda/sageattention/attn_utils.cuh
+++ b/src/core/cuda/sageattention/attn_utils.cuh
@@ -20,6 +20,7 @@
 #include <cuda_pipeline_primitives.h>
 
 #include "cp_async.cuh"
+#include "math.cuh"
 #include "mma.cuh"
 #include "permuted_smem.cuh"
 #include "numeric_conversion.cuh"