[package]
edition = "2021"
rust-version = "1.96"
name = "tenferro-einsum"
version = "0.3.0"
authors = [
"Hiroshi Shinaoka <h.shinaoka@gmail.com>",
"GiggleLiu <cacate0129@gmail.com>",
"Satoshi Terasaki <https://github.com/terasakisatoshi>",
]
build = false
publish = true
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Subscripts, contraction planning, concrete/traced/eager einsum APIs, extension runtime, and AD rule for tenferro."
homepage = "https://tensor4all.org/tenferro-rs/"
documentation = "https://docs.rs/tenferro-einsum"
readme = "README.md"
keywords = [
"tensor",
"einsum",
"contraction",
"numerical",
]
categories = [
"science",
"mathematics",
"algorithms",
]
license = "MIT OR Apache-2.0"
repository = "https://github.com/tensor4all/tenferro-rs"
[package.metadata.docs.rs]
rustdoc-args = [
"--cfg",
"docsrs",
]
features = [
"autodiff",
"cpu-faer",
"cuda",
"webgpu",
]
[features]
autodiff = [
"dep:tenferro-ad",
"tenferro-internal-ops/autodiff",
]
blas-accelerate = [
"tenferro-ad?/blas-accelerate",
"tenferro-cpu/blas-accelerate",
]
blas-mkl = [
"tenferro-ad?/blas-mkl",
"tenferro-cpu/blas-mkl",
]
blas-openblas = [
"tenferro-ad?/blas-openblas",
"tenferro-cpu/blas-openblas",
]
cpu-blas = [
"tenferro-ad?/cpu-blas",
"tenferro-cpu/cpu-blas",
]
cpu-faer = [
"tenferro-ad?/cpu-faer",
"tenferro-cpu/cpu-faer",
]
cuda = [
"tenferro-ad?/cuda",
"tenferro-internal-ops/cuda",
"dep:tenferro-gpu",
"tenferro-gpu/cuda",
]
default = ["cpu-faer"]
rocm = [
"tenferro-ad?/rocm",
"tenferro-internal-ops/rocm",
"dep:tenferro-gpu",
"tenferro-gpu/rocm",
]
webgpu = [
"tenferro-ad?/webgpu",
"tenferro-internal-ops/webgpu",
"dep:tenferro-gpu",
"tenferro-gpu/webgpu",
]
[lib]
name = "tenferro_einsum"
path = "src/lib.rs"
[[test]]
name = "integration"
path = "tests/integration.rs"
[[test]]
name = "prelude"
path = "tests/prelude.rs"
[[bench]]
name = "changing_shape_prepare"
path = "benches/changing_shape_prepare.rs"
harness = false
required-features = ["autodiff"]
[[bench]]
name = "einsum_cpu_bench"
path = "benches/einsum_cpu_bench.rs"
harness = false
required-features = ["autodiff"]
[[bench]]
name = "mps_inner_product"
path = "benches/mps_inner_product.rs"
harness = false
[[bench]]
name = "mps_inner_product_compile"
path = "benches/mps_inner_product_compile.rs"
harness = false
[dependencies.computegraph]
version = "0.1.0"
[dependencies.lru]
version = "0.12"
[dependencies.omeco]
version = "0.2.4"
[dependencies.smallvec]
version = "1"
[dependencies.tenferro-ad]
version = "0.3.0"
optional = true
default-features = false
[dependencies.tenferro-cpu]
version = "0.3.0"
default-features = false
[dependencies.tenferro-gpu]
version = "0.3.0"
optional = true
default-features = false
[dependencies.tenferro-internal-extension-macros]
version = "0.3.0"
[dependencies.tenferro-internal-ops]
version = "0.3.0"
default-features = false
[dependencies.tenferro-runtime]
version = "0.3.0"
default-features = false
[dependencies.tenferro-tensor]
version = "0.3.0"
default-features = false
[dependencies.thiserror]
version = "2"
[dev-dependencies.criterion]
version = "0.5"
[dev-dependencies.num-complex]
version = "0.4"
[dev-dependencies.tenferro-internal-ops]
version = "0.3.0"
default-features = false