kernels-community
/

mel_spectrogram

Model card Files Files and versions

mel_spectrogram / build.toml

drbh

feat: improve tests and bump build

916419b over 1 year ago

History Blame Contribute Delete

304 Bytes

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