dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
burn/crates/burn-ndarray
History
Genna Wingert 163a2069c6
Optimize connected components (#2810) 
* Add boolean and/or to bool tensors and fix version mismatch for rand

* Add tests

* Fix tch

* Fix for cubecl update

* Add missing test attribute

* Optimize NdArray into_data and other ways to make connected components faster

* Use existing into_data implementation

* Make lint `allow` because it's buggy

* `bool_into_data` should have a dtype of `BoolElem`

* Don't convert to bool in place

* Don't assert type equality for bool tests

* Fix nn/mask tests

* Fix and/or tests

* Add `to_bool` and fix `Bool::Elem`

* Fix display for bool

* Fix hardcoded `DType::Bool` uses
 		2025-02-17 09:51:38 -05:00
..
src Optimize connected components (#2810) 2025-02-17 09:51:38 -05:00
Cargo.toml Refactor/burn ir (#2796) 2025-02-11 14:29:14 -05:00
LICENSE-APACHE Update licenses symlinks (#1613) 2024-04-12 14:43:58 -04:00
LICENSE-MIT Update licenses symlinks (#1613) 2024-04-12 14:43:58 -04:00
README.md [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00
build.rs [refactor] Move burn crates to their own crates directory (#1336) 2024-02-20 13:57:55 -05:00

README.md

Burn NdArray

Burn ndarray backend

Current Crates.io Version license

Feature Flags

This crate can be used without the standard library (#![no_std]) with alloc by disabling the default std feature.

The following flags support various BLAS options:

  • blas-accelerate - Accelerate framework (macOS only)
  • blas-netlib - Netlib
  • blas-openblas - OpenBLAS static linked
  • blas-openblas-system - OpenBLAS from the system

Note: under the no_std mode, the seed is fixed if the seed is not initialized by by Backend::seed method.

Platform Support

Option CPU GPU Linux MacOS Windows Android iOS WASM
Pure Rust Yes No Yes Yes Yes Yes Yes Yes
Accelerate Yes No No Yes No No Yes No
Netlib Yes No Yes Yes Yes No No No
Openblas Yes No Yes Yes Yes Yes Yes No