[package]
name = "tensor-rs"
version = "0.5.9"
authors = ["yguan <yguan@example.com>"]
edition = "2021"
description = """
A typeless tensor library


"""
documentation = "https://docs.rs/tensor-rs"
homepage = "https://github.com/pipehappy1/auto-diff"
repository = "https://github.com/pipehappy1/auto-diff"
readme = "README.md"
license = "MIT"
keywords = ["machine-learning", "neural-network", "deep-learning"]
exclude = ["/dev/**"]

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
num-traits = "0.2"

rand = "0.8"
rand_distr = "0.4"

blas = { version = "0.22", optional = true }
blas-src = { version = "0.8", optional = true }
lapack = { version = "0.19", optional = true }
lapack-src = { version = "0.8", optional = true }

#rcublas = { version = "0.5", optional = true }
cuda11-cudart-sys = { version = "0.3", optional = true }
cuda11-cutensor-sys = { version = "0.3", optional = true }

serde = { version = "1.0", features = ["derive"], optional = true}

[dev-dependencies]
criterion = "0.3"

# for examples
csv = "1.1"

tensorboard-rs = { path = "../tensorboard-rs", version = "0.5.9"}

openblas-src = { version = "0.10" }

serde-pickle = {version = "0.6"}

[features]

default = ["use-f64", "use-blas-lapack", "use-serde"] #

use-f32 = []
use-f64 = []
use-usize = []
use-u8 = []

use-serde = ["serde"]

use-blas-lapack = ["blas", "blas-src", "lapack", "lapack-src"]

use-cuda = ["cuda11-cudart-sys", "cuda11-cutensor-sys"]