dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
936 Commits 50 Branches 17 Tags 89 MiB
Commit Graph

936 Commits

Author SHA1 Message Date
Louis Fortier-Dubois ab67b6b036
slice assign in candle (#1095) 2024-01-08 16:41:34 -05:00
Mykolas Peteraitis e132e21816
Small fix: documentation of Tensor::stack (#1105) 2024-01-08 08:38:27 -05:00
Sylvain Benner 3c056a41e3
[CI] Disable unused windows steps since wgpu tests are skipped (#1107) 2024-01-06 15:19:03 -05:00
Nathaniel Simard f0e10f1499
Refactor/fusion optimization (#1104) 2024-01-06 14:22:50 -05:00
Sylvain Benner 220ce251e0
Fix burn-compute cargo build command line in no-std (#1109) 
This allows to copy/paste the output of the cargo build command
line.
 2024-01-06 13:37:08 -05:00
Kirill Mavreshko 97297538b1
Remove _devauto fuctions (#518) (#1110) 2024-01-06 13:36:34 -05:00
Sylvain Benner fab344c565
[burn-compute] Add persistent cache for autotune in std environment (#1087) 
* Add new persistant cache to tune cache

* Serialize autotune presistent cache using vectors

* Properly load and save the persistent cachegf

* Print an error when autotune cache cannot be loaded

* Add tests for persistent cache

Use the same logic as the already implemented tests

* Cargo fmt

* Silence clippy check about implementing default for CliMetricsRenderer

* Add burn-compute feature flag autotune-persistent-cache

This allow burn-compute to remain no-std compliant

* debug

* Git ignore .dir-locals.el files

* Update documentation for compute_checksum implementation

* Expect messages should be an expectation not an  error message

* Replace silent eprintln! by log:warn! macro

* Remove clippy allow attribute

* Fix typos in documentation

* Move creation of additional client into the test fn that requires it

* Create compute clients in test function to test different checksum

* Revert tui as a default feature in burn-train cargo file

* Use structs for autotune cache entries

* Unpack InMemoryCacheEntry for even better readibility

* Remove uneeded checksum_checked field in no-std env

* Make sure that autotune cache directoy exists

* Add test for autotune cache file path creation

* Add prefix device info to autotune cache file

* Use new compute in autotune cache integration tests

This prevents race condition by always reloading the cache fir
each test.

* Move burn-compute rand depdencencie in dev-dependencies

* Avoid creation of formatted msg except in case of actual error

* Fix burn-compute unused code warning in no-std env
 2024-01-05 14:19:33 -05:00
Sylvain Benner b17ac2dfba
[CI] Pin version of CI dependencies from wgpu repo (#1120) 
* Pin version of CI dependencies from wgpu repo

Fixes seg fault in wgpu instance creation.

* Fix new_devauto dead code warning

* Add a notice for wgpu
 2024-01-05 12:47:31 -05:00
Sylvain Benner a4de93a39f
Fix import and type redefinitions in mnist example crate (#1100) 
* Remove the double import for WgpuDevice
* Prioritize wgpu backend over the default ndarray when wgpu feature is set

This fixes `cargo bench --festures wgpu` as `--no-default-features` cannot be
used.
 2024-01-02 12:47:44 -05:00
unrenormalizable 40ec289a92
Making Embedding.weight public (#1094) 2023-12-22 18:11:38 -05:00
Nathaniel Simard fceb036c6f
Fix: serde dependency (#1091) 
* Re-export serde

* Fix

* USe another strategy

* Fix

* Fix

* Update de book
 2023-12-22 16:53:34 -05:00
Nathaniel Simard b0a2b30ed1
Fix/wgpu/fusion (#1093) 
* Fix fusion

* Real fix

* Fix
 2023-12-22 16:53:19 -05:00
Louis Fortier-Dubois b070706310
Fix/wgpu/tanh (#1090) 2023-12-21 14:12:49 -05:00
Nathaniel Simard d82e6b157b
Fix tests (#1089) 
* Fix tests

* Fix fmt

* Fix CI
 2023-12-21 13:06:19 -05:00
Kirill Mavreshko 1fd07fcb4a
Explicit device tensors (#1081) 2023-12-20 17:49:59 -05:00
Kelvin Wu 7c6f017c98
Implement chunk for different backends (#1032) 2023-12-20 13:35:59 -05:00
pacowong c1cb77ac2e
Add mish and softplus activation function (#1071) 2023-12-20 12:41:23 -05:00
Louis Fortier-Dubois 4fb5b74087
fix multithread progress aggregation in dataloader (#1083) 2023-12-19 16:28:57 -05:00
Nathaniel Simard 75062c51e0
Refactor/wgpu/binary (#1078) 
* Refactor binary

* Fix

* Oups

* Preparation for cmp

* Refactor comparison

* Remove templates

* Cleanup

* Cleanup

* Fix typo

* Code review
 2023-12-19 14:44:55 -05:00
Nathaniel Simard b5c49c5bf7
Fusion wgpu compilation cache (#1069) 
* Refactor fusion in the wgpu backend

* WIP

* Refactor

* WIP

* Fix inplace ops

* Works ish

* Cleanup Output

* Refactoring

* Refactor Clamp

* Cleanup

* Cleanup

* Updates

* Fix CI

* Code review
 2023-12-18 12:16:08 -05:00
Nathaniel Simard 042454a9db
Example: Update Text Classification (#1044) 
* Example: Update Text Classification

* cargo fmt

---------

Co-authored-by: louisfd <louisfd94@gmail.com>
 2023-12-18 08:42:07 -05:00
Louis Fortier-Dubois 3f06e81f2d
Add stride field to reduce key (#1070) 2023-12-15 13:35:46 -05:00
Luni-4 467e65247b ci/Add Windows pipeline (#925) 
Co-Authored-By: Sylvain Benner <sylvain.benner@gmail.com>
 2023-12-14 18:48:59 -05:00
AuruTus fc97a28f16
fix/ndarray: remove reversed axes check (#1058) 2023-12-14 18:09:24 -05:00
Louis Fortier-Dubois 4608cd97d2
Feat/wgpu/repeat (#1068) 2023-12-14 18:08:37 -05:00
Nathaniel Simard 2ca1fd5a92
Fix bugs (#1064) 2023-12-12 16:19:01 -05:00
Jantxu 70eda6b34e
use web-time in case of wasm (#1060) 2023-12-12 13:47:10 -05:00
Alex Errant 610d64095e
cargo +nightly fmt (#1017) 2023-12-12 13:29:06 -05:00
Luni-4 1a5f252ac6
Ci/Improve scheduled tasks (#1028) 2023-12-12 09:42:33 -05:00
David Chavez 71d3c1d142
chore(infra): Share some properties across workspace (#1039) 2023-12-12 09:39:07 -05:00
David Chavez 3066196297
Fix where sqrt should’ve been powf(2.) (#1052) 2023-12-06 19:02:15 -05:00
David Chavez ef09b637b9
chore(web): better panic (#1051) 2023-12-05 07:30:29 -05:00
David Chavez d9f93d31b4
test burn-ndarray also with accelerate (#1050) 2023-12-04 13:19:46 -05:00
David Chavez 82f5722ca1
chore(ci): CI grouping and refactoring (#1024) 2023-12-04 11:14:50 -05:00
Louis Fortier-Dubois 8fc52113bc
Chore/bump v12 (#1048) 2023-12-04 10:47:54 -05:00
Louis Fortier-Dubois 3088c466a5
patch 0.11.1 (#1047) 2023-12-04 10:18:30 -05:00
Nathaniel Simard 0b16786729
Fix/orphan cleanup (#1043) 2023-12-04 09:44:17 -05:00
Nathaniel Simard 9d92f9a7b8
Update readme (#1045) 2023-12-04 09:39:55 -05:00
Timothy 52811f9938
Fix some issues in Burn book (#1042) 2023-12-04 09:23:47 -05:00
David Chavez cb8d4dac62
Fix building docs (#1040) 2023-12-02 12:08:28 -05:00
nathaniel 3a8dcfb5f4 Fix burn-train feature flag 2023-12-01 16:01:22 -05:00
Louis Fortier-Dubois 4d7944912d
fix gatter typo (#1036) 2023-12-01 15:28:25 -05:00
Nathaniel Simard ab1b5890f5
Chore/release (#1031) 2023-12-01 14:33:28 -05:00
Dilshod Tadjibaev 4192490b88
Update with demo links (#1033) 2023-12-01 13:07:13 -05:00
Nathaniel Simard 12911c3ad6
Fix/mnist inference web (#1029) 2023-12-01 12:05:31 -05:00
Nathaniel Simard 670280dda2
Feat/fusion/cache (#1020) 2023-12-01 12:05:11 -05:00
Nathaniel Simard b0de56da29
Fix double broadcast with tch (#1026) 
* Fix double broadcast with tch

* More fixes

* Fix clippy warm
 2023-12-01 10:02:57 -05:00
Nathaniel Simard f6d14f1b1a
Refactor feature flags (#1025) 2023-12-01 09:48:28 -05:00
wcshds 1f18cf4adf
implement DoubleEndedIterator trait for DimIter (#1022) 
* implement DoubleEndedIterator for DimIter

* return early in next() and next_back()

* return early in next() and next_back()

* remove #[rustfmt::skip]
 2023-11-30 12:26:16 -06:00
Dilshod Tadjibaev 5e4e0b6bb3
Change ONNX default record type to NamedMpk (#1019) 
* Change ONNX default record type to NamedMpk

This temporarily workaround of #952 bug.

* Fix formatting

* Fix URL in the doc
 2023-11-30 13:22:34 -05:00