[package]
edition = "2024"
name = "luma-tensor"
version = "0.0.1"
build = false
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "tensor crate with autograd"
readme = "README.md"
keywords = [
"machine-learning",
"auto-grad",
]
license = "MIT"
repository = "https://github.com/MoleSir/luma.git"
resolver = "1"
[features]
cuda = [
"dep:cudarc",
"dep:luma-cuda-kernel",
]
default = []
[lib]
name = "luma_tensor"
path = "src/lib.rs"
[[example]]
name = "cuda"
path = "examples/cuda.rs"
[[test]]
name = "cpu"
path = "tests/cpu.rs"
[[test]]
name = "cuda"
path = "tests/cuda.rs"
[dependencies.cudarc]
version = "0.19.8"
features = [
"cuda-version-from-build-system",
"driver",
"nvrtc",
"cublas",
]
optional = true
[dependencies.gemm]
version = "0.19"
[dependencies.libm]
version = "0.2.15"
[dependencies.luma-cuda-kernel]
version = "0.0.1"
optional = true
package = "luma-cuda-kernel"
[dependencies.num_cpus]
version = "1.17"
[dependencies.paste]
version = "1.0.15"
[dependencies.rand]
version = "0.9.2"
[dependencies.rand_distr]
version = "0.5.1"
[dependencies.thiserror]
version = "2.0"