[general]
name = "optimizer"
backends = [
    "cuda",
    "rocm",
]

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

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

[kernel.optimizer_rocm]
backend = "rocm"
rocm-archs = [
    "gfx906",
    "gfx908",
    "gfx90a",
    "gfx940",
    "gfx941",
    "gfx942",
    "gfx1030",
    "gfx1100",
    "gfx1101",
]
depends = ["torch"]
src = ["optimizer/dummy.cu"]