From 6edeac1b2061eec8abb021a7b534b899b32dc93c Mon Sep 17 00:00:00 2001
From: leejet <leejet714@gmail.com>
Date: Sun, 1 Nov 2026 13:39:29 +0800
Subject: [PATCH] fix: include math helpers directly in SageAttention utilities

---
 src/core/cuda/sageattention/attn_utils.cuh | 1 +
 1 file changed, 1 insertion(+)

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"