# THIS FILE IS AUTOMATICALLY GENERATED BY CARGO
#
# When uploading crates to the registry Cargo will automatically
# "normalize" Cargo.toml files for maximal compatibility
# with all versions of Cargo and also rewrite `path` dependencies
# to registry (e.g., crates.io) dependencies.
#
# If you are reading this file be aware that the original Cargo.toml
# will likely look very different (and much more reasonable).
# See Cargo.toml.orig for the original contents.

[package]
edition = "2021"
rust-version = "1.65"
name = "dfdx"
version = "0.13.0"
description = "Ergonomic auto differentiation in Rust, with pytorch like apis."
homepage = "https://github.com/coreylowman/dfdx"
documentation = "https://docs.rs/dfdx"
readme = "README.md"
keywords = [
    "deep-learning",
    "neural-network",
    "backprop",
    "tensor",
    "autodiff",
]
license = "MIT OR Apache-2.0"
repository = "https://github.com/coreylowman/dfdx"

[package.metadata.docs.rs]
features = [
    "nightly",
    "numpy",
    "safetensors",
    "cuda",
    "ci-check",
]

[[bench]]
name = "batchnorm2d"
harness = false

[[bench]]
name = "conv2d"
harness = false

[[bench]]
name = "sum"
harness = false

[[bench]]
name = "softmax"
harness = false

[dependencies.cudarc]
version = "0.9.13"
features = [
    "driver",
    "cublas",
    "nvrtc",
]
optional = true
default-features = false

[dependencies.gemm]
version = "0.15.4"
optional = true
default-features = false

[dependencies.half]
version = "2.3.1"
features = [
    "num-traits",
    "rand_distr",
]
optional = true

[dependencies.libm]
version = "0.2.7"

[dependencies.memmap2]
version = "0.5"
optional = true
default-features = false

[dependencies.no-std-compat]
version = "0.4.1"
features = [
    "alloc",
    "compat_hash",
]
optional = true
default-features = false

[dependencies.num-traits]
version = "0.2.15"
default-features = false

[dependencies.rand]
version = "0.8.5"
features = ["std_rng"]
default-features = false

[dependencies.rand_distr]
version = "0.4.3"
default-features = false

[dependencies.rayon]
version = "1.7.0"
optional = true

[dependencies.safetensors]
version = "0.3"
optional = true
default-features = false

[dependencies.spin]
version = "0.9.8"
features = [
    "spin_mutex",
    "rwlock",
    "portable_atomic",
]
optional = true
default-features = false

[dependencies.zip]
version = "0.6.6"
optional = true
default-features = false

[dev-dependencies.indicatif]
version = "0.17.3"

[dev-dependencies.mnist]
version = "0.5.0"

[dev-dependencies.tempfile]
version = "3.3.0"

[build-dependencies.glob]
version = "0.3.1"
optional = true

[features]
ci-check = ["cudarc?/ci-check"]
cpu = [
    "dep:gemm",
    "dep:rayon",
]
cuda = [
    "dep:cudarc",
    "dep:glob",
]
cudnn = [
    "cuda",
    "cudarc?/cudnn",
]
default = [
    "std",
    "fast-alloc",
    "cpu",
]
f16 = [
    "dep:half",
    "cudarc?/f16",
]
fast-alloc = ["std"]
nightly = [
    "half?/use-intrinsics",
    "gemm?/nightly",
]
no-std = [
    "no-std-compat",
    "dep:spin",
    "cudarc?/no-std",
    "num-traits/libm",
]
numpy = [
    "dep:zip",
    "std",
]
safetensors = [
    "dep:safetensors",
    "std",
    "dep:memmap2",
]
std = [
    "cudarc?/std",
    "rand_distr/std_math",
    "gemm?/std",
]
test-amp-f16 = ["f16"]
test-f16 = ["f16"]
test-f64 = []
test-integrations = []