[general]
name = "residual_rms_rocm"
universal = false

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

[kernel.residual_rms_rocm]
depends = ["torch"]
backend = "rocm"
rocm-archs = [
    "gfx908",
    "gfx90a",
    "gfx940",
    "gfx941",
    "gfx942",
    "gfx1100",
    "gfx1101",
    "gfx1102",
    "gfx1200",
    "gfx1201",
]
src = [
    "residual_rms_rocm/residual_rms_dispatch.cu",
    "residual_rms_rocm/residual_rms_scalar.cu",
    "residual_rms_rocm/residual_rms_vectorized.cu",
    "residual_rms_rocm/utils.h",
]
include = ["."]