File size: 1,151 Bytes
c148255 | 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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | [general]
backends = [
"cpu",
"cuda",
"metal",
"rocm",
"xpu",
]
name = "__KERNEL_NAME__"
version = 1
[torch]
src = [
"torch-ext/torch_binding.cpp",
"torch-ext/torch_binding.h",
]
[kernel.__KERNEL_NAME_NORMALIZED__]
backend = "cuda"
depends = ["torch"]
src = ["__KERNEL_NAME_NORMALIZED___cuda/__KERNEL_NAME_NORMALIZED__.cu"]
[kernel.__KERNEL_NAME_NORMALIZED___metal]
backend = "metal"
depends = ["torch"]
src = [
"__KERNEL_NAME_NORMALIZED___metal/__KERNEL_NAME_NORMALIZED__.mm",
"__KERNEL_NAME_NORMALIZED___metal/__KERNEL_NAME_NORMALIZED__.metal",
]
[kernel.__KERNEL_NAME_NORMALIZED___rocm]
backend = "rocm"
depends = ["torch"]
rocm-archs = [
"gfx906",
"gfx908",
"gfx90a",
"gfx940",
"gfx941",
"gfx942",
"gfx1030",
"gfx1100",
"gfx1101",
]
src = ["__KERNEL_NAME_NORMALIZED___cuda/__KERNEL_NAME_NORMALIZED__.cu"]
[kernel.__KERNEL_NAME_NORMALIZED___xpu]
backend = "xpu"
depends = ["torch"]
src = ["__KERNEL_NAME_NORMALIZED___xpu/__KERNEL_NAME_NORMALIZED__.cpp"]
[kernel.__KERNEL_NAME_NORMALIZED___cpu]
backend = "cpu"
depends = ["torch"]
src = ["__KERNEL_NAME_NORMALIZED___cpu/__KERNEL_NAME_NORMALIZED___cpu.cpp"]
|