From 62c54439c8bb4da32cab1d81981e7d447e2c93f3 Mon Sep 17 00:00:00 2001
From: leejet <leejet714@gmail.com>
Date: Mon, 2 Nov 2026 00:17:49 +0800
Subject: [PATCH] fix: include missing cassert headers in SageAttention CUDA
 code

---
 src/core/cuda/sageattention/mma.cuh                | 1 +
 src/core/cuda/sageattention/numeric_conversion.cuh | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/core/cuda/sageattention/mma.cuh b/src/core/cuda/sageattention/mma.cuh
index 4de48286..f473fa72 100644
--- a/src/core/cuda/sageattention/mma.cuh
+++ b/src/core/cuda/sageattention/mma.cuh
@@ -22,6 +22,7 @@
 #include <cuda_bf16.h>
 #include <cuda_fp16.h>
 #include <cuda_runtime.h>
+#include <cassert>
 #include <type_traits>
 
 namespace sageattention {
diff --git a/src/core/cuda/sageattention/numeric_conversion.cuh b/src/core/cuda/sageattention/numeric_conversion.cuh
index 066f7fe0..8da67d5f 100644
--- a/src/core/cuda/sageattention/numeric_conversion.cuh
+++ b/src/core/cuda/sageattention/numeric_conversion.cuh
@@ -22,6 +22,7 @@
 #include <cuda_fp16.h>
 #include <cuda_fp8.h>
 #include <cuda_runtime.h>
+#include <cassert>
 
 namespace sageattention {