build.toml

[general]
name = "relu_metal"
universal = false

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

[kernel.activation]
backend = "metal"
src = [
  "relu/relu.mm",
]
depends = [ "torch" ]