[general]
name = "model-as-a-kernel"
version = 1
edition = 5
license = "Apache-2.0"
backends = [
  "cuda",
]

[general.hub]
repo-id = "phanerozoic/model-as-a-kernel"

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

[kernel.model-as-a-kernel]
backend = "cuda"
cuda-capabilities = ["8.0", "8.6", "8.9", "9.0", "10.0", "12.0"]
depends = ["torch"]
src = [
  "mak_cuda/megakernel.cu",
  "mak_cuda/mak_kernels.h",
]