From 4c194b7254b5f8fffd09a6784517cfa747bc31e7 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 05:15:48 +0000
Subject: [PATCH] fix: add CUDA include path for host compiler

---
 tests/unit/test_fmha_sm100.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/unit/test_fmha_sm100.py b/tests/unit/test_fmha_sm100.py
index 879719d8..c084e939 100644
--- a/tests/unit/test_fmha_sm100.py
+++ b/tests/unit/test_fmha_sm100.py
@@ -75,6 +75,9 @@ try:
             f"-I{REPO}",
             f"-I{CUTLASS}/include",
         ],
+        extra_cflags=[
+            f"-I/usr/local/cuda-13.2/include",
+        ],
         verbose=True,
     )
     print("\n✅ JIT compilation PASSED!")