[package]
edition = "2021"
name = "ferrox-cuda"
version = "0.20.0"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "CUDA/NVRTC kernels for the Ferrox inference engine"
homepage = "https://github.com/antonellof/ferrox"
readme = "README.md"
keywords = [
"cuda",
"gpu",
"llm",
"nvidia",
"ferrox",
]
categories = [
"hardware-support",
"science",
]
license = "Apache-2.0"
repository = "https://github.com/antonellof/ferrox"
[features]
cuda = ["dep:cudarc"]
default = []
[lib]
name = "ferrox_cuda"
path = "src/lib.rs"
[[example]]
name = "mul_mm_host_check"
path = "examples/mul_mm_host_check.rs"
[dependencies.cudarc]
version = "=0.11.9"
features = [
"std",
"driver",
"nvrtc",
"cuda-11080",
]
optional = true
default-features = false
[dependencies.thiserror]
version = "1"
[dev-dependencies.ferrox-quant]
version = "0.20.0"
[dev-dependencies.half]
version = "2"