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-router
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
README.md Add `BackendRouter` to handle multiple backends (#2353) 2024-10-18 14:23:26 -04:00

README.md

Burn Router

A multi-backend extension that forwards the tensor operations to the appropriate backend.