[general]
name = "rotary"
universal = false

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

[kernel.activation]
backend = "cuda"
depends = ["torch"]
src = ["rotary/rotary_cuda.cu"]

[kernel.rotary_xpu]
backend = "xpu"
depends = ["torch"]
src = [
    "rotary-xpu/rotary_xpu.cpp",
    "rotary-xpu/rotary_xpu.hpp",
    ]