my_softmax_function / build.toml
tiagomonteiro0715
migrate to new huggingface/kernels builder schema
fcfe221
Raw
History Blame Contribute Delete
474 Bytes
# Hugging Face kernel manifest (new huggingface/kernels schema).
# Pure-Python kernel (CUTLASS Python DSL, JIT-compiled on the GPU at runtime):
# no compiled sources -> declared with [torch-noarch]. CUDA-only (CUTLASS is NVIDIA).
[general]
name = "my-softmax-function"
version = 1
edition = 5
license = "Apache-2.0"
backends = ["cuda"]
[general.cuda]
# nvidia-cutlass-dsl 4.3.0 requires CUDA >= 12.8.
minver = "12.8"
python-depends = ["nvidia-cutlass-dsl"]
[torch-noarch]