build.toml

[general]
name = "mel_spectrogram"

[torch]
src = [
  "torch-ext/torch_binding.cpp",
  "torch-ext/torch_binding.h"
]

[kernel.mel_spectrogram]
cuda-capabilities = [ 
   # "7.0", "7.2", "7.5", 
   "8.0", "8.6", "8.7", "8.9", "9.0" 
]
src = [
  "mel_spectrogram/mel_spectrogram.cu"
]
depends = [ "torch" ]