build.toml

[general]
name = "rmsnorm_kernel"

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

[kernel.rmsnorm_kernel]
src = [
  "rmsnorm_kernel/rmsnorm.cu",
]
depends = [ "torch"]
cuda-capabilities = [ "12.3" ]