File size: 777 Bytes
84ec9f0 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | [general]
name = "torch_harmonics_attn"
universal = false
[torch]
src = [
"torch-ext/torch_binding.cpp",
"torch-ext/torch_binding.h",
]
[kernel.torch_harmonics_attn]
depends = ["torch"]
backend = "cuda"
cuda-capabilities = [
"7.5",
"8.0",
"8.9",
"9.0",
"10.0",
]
src = [
"torch_harmonics_attn/attention_cpu_bwd.cpp",
"torch_harmonics_attn/attention_cpu_fwd.cpp",
"torch_harmonics_attn/attention_cpu.h",
"torch_harmonics_attn/attention_cuda_bwd.cu",
"torch_harmonics_attn/attention_cuda_fwd.cu",
"torch_harmonics_attn/attention_cuda_utils.cu",
"torch_harmonics_attn/attention_cuda_utils.cuh",
"torch_harmonics_attn/attention_cuda.cuh",
"torch_harmonics_attn/attention.h",
"torch_harmonics_attn/cudamacro.h"
]
|