Lib.rs

Cargo Features

[dependencies]
burn-tensor = { version = "0.16.0", default-features = false, features = ["std", "cubecl", "cubecl-cuda", "cubecl-hip", "cubecl-wgpu", "experimental-named-tensor", "export_tests", "repr", "doc"] }

default doc? = repr, std

These default features are set whenever burn-tensor is added without default-features = false somewhere in the dependency tree.

Enables rayon of burn-common

std default = colored

Enables std of burn-common, half, num-traits, and rand

rand:

Default enables std

cubecl cubecl-cuda? cubecl-hip? cubecl-wgpu? export_tests?

Enables cubecl

cubecl-cuda = cubecl

Enables cuda of cubecl

cubecl:

Runtimes

cubecl-hip = cubecl

Enables hip of cubecl

cubecl-wgpu = cubecl

Enables wgpu of cubecl

experimental-named-tensor

export_tests = burn-tensor-testgen, cubecl

Affects burn-tensor::tests…

repr default

Affects burn-tensor::repr…

doc = default

Features from optional dependencies

In crates that don't use the dep: syntax, optional dependencies automatically become Cargo features. These features may have been created by mistake, and this functionality may be removed in the future.

burn-tensor-testgen export_tests?

colored std

Enables colored ^2.1.0