Added new attention_utils.cu source.

46bd1b26 · Mauro Bisson · Thorsten Kurth · 9a463332 · 46bd1b26
Commit 46bd1b26 authored Jul 10, 2025 by Mauro Bisson Committed by Thorsten Kurth Jul 15, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

setup.py setup.py +1 -0

No files found.
--- a/setup.py
+++ b/setup.py
@@ -102,6 +102,7 @@ def get_ext_modules():
            CUDAExtension(
                name="attention_cuda_extension",
                sources=[
+                    "torch_harmonics/csrc/attention/attention_utils.cu",
                    "torch_harmonics/csrc/attention/attention_fwd_cuda.cu",
                    "torch_harmonics/csrc/attention/attention_bwd_cuda.cu",
                    "torch_harmonics/csrc/attention/attention_interface.cu",