[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"]