From 09dfd4a41f10be1432ed7d136e47f30c778fcaca Mon Sep 17 00:00:00 2001 From: biondizzle Date: Thu, 28 May 2026 05:16:41 +0000 Subject: [PATCH] fix: rename .cpp to .cu for CUDA compilation --- dsv4/kernels/attention/{fmha_sm100.cpp => fmha_sm100_launch.cu} | 0 tests/unit/test_fmha_sm100.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) rename dsv4/kernels/attention/{fmha_sm100.cpp => fmha_sm100_launch.cu} (100%) 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}",