[general]
name = "rmsnorm_metal"
universal = false

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

[kernel.rmsnorm_metal]
backend = "metal"
src = [
  "",
  "rmsnorm_metal/rmsnorm.metal",
  "rmsnorm_metal/rmsnorm.mm",
]
depends = [ "torch" ]