Skip to content

GitLab

Commit 46bd1b26 authored by Mauro Bisson's avatar Mauro Bisson Committed by Thorsten Kurth
Browse files

Added new attention_utils.cu source.

parent 9a463332
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
setup.py
View file @ 46bd1b26
...@@ -102,6 +102,7 @@ def get_ext_modules(): ...@@ -102,6 +102,7 @@ def get_ext_modules():
CUDAExtension( CUDAExtension(
name="attention_cuda_extension", name="attention_cuda_extension",
sources=[ sources=[
"torch_harmonics/csrc/attention/attention_utils.cu",
"torch_harmonics/csrc/attention/attention_fwd_cuda.cu", "torch_harmonics/csrc/attention/attention_fwd_cuda.cu",
"torch_harmonics/csrc/attention/attention_bwd_cuda.cu", "torch_harmonics/csrc/attention/attention_bwd_cuda.cu",
"torch_harmonics/csrc/attention/attention_interface.cu", "torch_harmonics/csrc/attention/attention_interface.cu",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment