Lib.rs

Cargo Features

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

default doc? = repr, std

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

std default = colored

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

rand:

Default enables std

experimental-named-tensor

export_tests = burn-tensor-testgen

repr default

Affects burn-tensor::repr…

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

Enables cubecl

cubecl-wgpu = cubecl

Enables wgpu of cubecl

cubecl-cuda = cubecl

Enables cuda of cubecl

cubecl:

Runtimes

cubecl-hip = cubecl

Enables hip of cubecl

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