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!")