[general]
name = "spec-decode-ops"
version = 1
edition = 5
license = "Apache-2.0"
backends = [
  "cuda",
]

[general.hub]
repo-id = "phanerozoic/spec-decode-ops"

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

[kernel.spec-decode-ops]
backend = "cuda"
cuda-capabilities = ["8.0", "8.6", "8.9", "9.0"]
depends = ["torch"]
src = ["spec_decode_cuda/spec_decode_ops.cu"]