build.toml

[general]
name = "fused-split"
license = "Apache-2.0"
backends = [
  "rocm",
]
version = 1
edition = 5

[general.hub]
repo-id = "Ashiedu/fused-split"

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

[kernel.fused_split_rocm]
backend = "rocm"
depends = ["torch"]
rocm-archs = [
  "gfx1030",
]
src = ["fused_split_cuda/fused_split.cu"]