[general]
name = "rmsnorm_kernel"
universal = false

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

[kernel.rmsnorm_kernel]
backend = "cuda"
src = [
  "rmsnorm_kernel/rmsnorm.cu",
]
depends = [ "torch"]