build.toml

[general]
name = "optimizer"
universal = false

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

[kernel.activation]
backend = "rocm"
src = [
  "optimizer/dummy.cu",
]
depends = [ "torch" ]

[kernel.activation_cuda]
backend = "cuda"
src = [
  "optimizer/dummy.cu",
]
depends = [ "torch" ]