fix: --x cu for .cuh compilation

2026-05-28 05:06:13 +00:00
parent 4dcd80ea0d
commit 66d9f5c60f
1 changed files with 1 additions and 0 deletions
--- a/tests/unit/test_fmha_sm100.py
+++ b/tests/unit/test_fmha_sm100.py
@@ -39,6 +39,7 @@ nvcc_cmd = [
    "-DCUTE_ARCH_TCGEN05_F16F32_MMA_ENABLED",
    "-c",
    f"{REPO}/dsv4/kernels/attention/fmha_sm100.cuh",
+    "--x", "cu",
    "-o", "/tmp/fmha_sm100_test.o",
    "--ptxas-options=-v",
 ]