From 66d9f5c60fc611c5f09b2fa76a886ef2c669654b Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 05:06:13 +0000
Subject: [PATCH] fix: --x cu for .cuh compilation

---
 tests/unit/test_fmha_sm100.py | 1 +
 1 file changed, 1 insertion(+)

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",
 ]