[general]
version = "0.0.1"

[torch]
name = "mel_spectrogram"
src = [
  "ext-torch/registration.h",
  "ext-torch/torch_binding.cpp",
  "ext-torch/torch_binding.h",
]
pyroot = "ext-torch"

[kernel.mel_spectrogram]
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"]