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
Guillaume Lagrange eb899db16c
Add ops w/ default implementation for `QTensorOps` (#2125) 
* Add q_* ops to match float ops

* Refactor q_* ops w/ dequant_op_quant macro

* Comparison ops are already implemented by default to compare dequantized values

* Add default arg min/max implementation and fix tch implementation

* Avoid division by zero scale

* Add default q_gather implementation (tch does not support on quantized tensor)

* Add warning instead for tch quantize_dynamic

* Call chunk backend implementation

* Add QFloat check for q_ ops

* Add tch q_min/max_dim_with_indices

* Add q_ ops tests

* Clippy fix

* Remove dead code/comments

* Fix quantization tests precision

* Set higher tolerance for ndarray backend

* Remove comment
 		2024-09-09 12:21:47 -04:00
..
src Add ops w/ default implementation for `QTensorOps` (#2125) 2024-09-09 12:21:47 -04:00
Cargo.toml Bump burn version to 0.15.0 2024-08-27 15:13:40 -04: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