[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"
]