[package]
edition = "2024"
name = "kvbm-kernels"
version = "1.4.0"
authors = ["NVIDIA Inc. <sw-dl-dynamo@nvidia.com>"]
build = "build.rs"
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Dynamo Inference Framework"
readme = "README.md"
license = "Apache-2.0"
repository = "https://github.com/ai-dynamo/dynamo.git"
[features]
default = []
kvbench = ["dep:clap"]
permute_kernels = []
static-kernels = []
testing-cuda = []
[lib]
name = "kvbm_kernels"
crate-type = [
"rlib",
"cdylib",
]
path = "src/lib.rs"
[[example]]
name = "kvbench"
path = "examples/kvbench.rs"
required-features = ["kvbench"]
[[test]]
name = "kernel_roundtrip"
path = "tests/kernel_roundtrip.rs"
[[test]]
name = "memcpy_batch"
path = "tests/memcpy_batch.rs"
[[test]]
name = "stub_build"
path = "tests/stub_build.rs"
[dependencies.clap]
version = "4"
features = ["derive"]
optional = true
[dependencies.cudarc]
version = "=0.19.8"
features = [
"cuda-version-from-build-system",
"fallback-latest",
]
[dev-dependencies.cudarc]
version = "=0.19.8"
features = [
"cuda-version-from-build-system",
"fallback-latest",
"f16",
]
[dev-dependencies.half]
version = "2"
[dev-dependencies.ndarray]
version = "0.17.2"
[dev-dependencies.rand]
version = "0.9.2"
[build-dependencies]