fix: rename .cpp to .cu for CUDA compilation

2026-05-28 05:16:41 +00:00
parent 4c194b7254
commit 09dfd4a41f
2 changed files with 1 additions and 1 deletions
--- a/dsv4/kernels/attention/fmha_sm100_launch.cu
+++ b/dsv4/kernels/attention/fmha_sm100_launch.cu
--- 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}",