diff --git a/dsv4/kernels/attention/fmha_sm100.cpp b/dsv4/kernels/attention/fmha_sm100_launch.cu
similarity index 100%
rename from dsv4/kernels/attention/fmha_sm100.cpp
rename to dsv4/kernels/attention/fmha_sm100_launch.cu
diff --git a/tests/unit/test_fmha_sm100.py b/tests/unit/test_fmha_sm100.py
index c084e939..5cd6b6a3 100644
--- a/tests/unit/test_fmha_sm100.py
+++ b/tests/unit/test_fmha_sm100.py
@@ -69,7 +69,7 @@ try:
 
     module = load(
         name="fmha_sm100",
-        sources=[f"{REPO}/dsv4/kernels/attention/fmha_sm100.cpp"],
+        sources=[f"{REPO}/dsv4/kernels/attention/fmha_sm100_launch.cu"],
         extra_cuda_cflags=[
             "-gencode=arch=compute_100a,code=sm_100a",
             f"-I{REPO}",