[package]
edition = "2024"
name = "burn-dispatch"
version = "0.21.0-pre.2"
authors = [
"laggui <lagrange.guillaume.1@gmail.com>",
"nathanielsimard <nathaniel.simard.42@gmail.com>",
]
build = "build.rs"
autolib = false
autobins = false
autoexamples = false
autotests = false
autobenches = false
description = "Backend dispatch for the Burn framework"
documentation = "https://docs.rs/burn-dispatch"
readme = "README.md"
keywords = [
"deep-learning",
"machine-learning",
"data",
]
categories = ["science"]
license = "MIT OR Apache-2.0"
repository = "https://github.com/tracel-ai/burn/tree/main/crates/burn-dispatch"
resolver = "2"
[package.metadata.docs.rs]
features = ["doc"]
rustdoc-args = [
"--cfg",
"docsrs",
]
[features]
autodiff = ["burn-autodiff"]
cpu = ["burn-cpu"]
cuda = ["burn-cuda"]
default = [
"std",
"ndarray",
"burn-autodiff?/default",
"burn-cpu?/default",
"burn-cuda?/default",
"burn-ndarray?/default",
"burn-rocm?/default",
"burn-tch?/default",
"burn-wgpu?/default",
]
doc = ["default"]
metal = [
"wgpu",
"burn-wgpu/metal",
]
ndarray = ["burn-ndarray"]
rocm = ["burn-rocm"]
std = [
"burn-backend/std",
"burn-std/std",
"burn-autodiff?/std",
"burn-cpu?/std",
"burn-cuda?/std",
"burn-ndarray?/std",
"burn-rocm?/std",
"burn-tch?/std",
"burn-wgpu?/std",
]
tch = ["burn-tch"]
tracing = [
"burn-autodiff?/tracing",
"burn-cpu?/tracing",
"burn-cuda?/tracing",
"burn-ndarray?/tracing",
"burn-rocm?/tracing",
"burn-tch?/tracing",
"burn-wgpu?/tracing",
]
vulkan = [
"wgpu",
"burn-wgpu/vulkan",
]
webgpu = [
"wgpu",
"burn-wgpu/webgpu",
]
wgpu = ["burn-wgpu"]
[lib]
name = "burn_dispatch"
path = "src/lib.rs"
[dependencies.burn-autodiff]
version = "=0.21.0-pre.2"
optional = true
default-features = false
[dependencies.burn-backend]
version = "=0.21.0-pre.2"
default-features = false
[dependencies.burn-cpu]
version = "=0.21.0-pre.2"
optional = true
default-features = false
[dependencies.burn-cuda]
version = "=0.21.0-pre.2"
optional = true
default-features = false
[dependencies.burn-ndarray]
version = "=0.21.0-pre.2"
optional = true
default-features = false
[dependencies.burn-rocm]
version = "=0.21.0-pre.2"
optional = true
default-features = false
[dependencies.burn-std]
version = "=0.21.0-pre.2"
default-features = false
[dependencies.burn-tch]
version = "=0.21.0-pre.2"
optional = true
default-features = false
[dependencies.burn-wgpu]
version = "=0.21.0-pre.2"
optional = true
default-features = false
[dependencies.paste]
version = "1"
[lints.clippy]
[lints.rustdoc]
broken_intra_doc_links = "deny"
invalid_html_tags = "deny"