diff --git a/tests/unit/test_fmha_sm100.py b/tests/unit/test_fmha_sm100.py
index 3ad241f5..c24de076 100644
--- 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",
 ]