fix: add CUDA include path for host compiler

2026-05-28 05:15:48 +00:00
parent 48baea7728
commit 4c194b7254
1 changed files with 3 additions and 0 deletions
--- 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!")