[general]
name = "logits-processor"
version = 1
edition = 5
license = "Apache-2.0"
backends = [
  "cuda",
]

[general.hub]
repo-id = "phanerozoic/logits-processor"

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

[kernel.logits_processor]
backend = "cuda"
cuda-capabilities = ["8.0", "8.6", "8.9", "9.0", "10.0", "12.0"]
depends = ["torch"]
src = ["logits_processor_cuda/logits_processor.cu"]