fix: include cuda_bf16.h unconditionally, add --expt-relaxed-constexpr

2026-05-28 05:13:01 +00:00
parent c1266b5275
commit 6bd3356582
2 changed files with 2 additions and 9 deletions
--- a/tests/unit/test_fmha_sm100.py
+++ b/tests/unit/test_fmha_sm100.py
@@ -42,6 +42,7 @@ nvcc_cmd = [
    "--x", "cu",
    "-o", "/tmp/fmha_sm100_test.o",
    "--ptxas-options=-v",
+    "--expt-relaxed-constexpr",
 ]

 print(f"nvcc command: {' '.join(nvcc_cmd)}")