File size: 478 Bytes
8535e80 ae32572 8535e80 ae32572 8535e80 ae32572 8535e80 ae32572 8535e80 ae32572 8535e80 ae32572 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
[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"]
|