dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
692 Commits 39 Branches 19 Tags 101 MiB
Commit Graph

692 Commits

Author SHA1 Message Date
Zsombor 4fc0c27e31
Implement tensor.recip() function to calculate elementwise reciprocals (#953) 2023-11-15 09:17:32 -05:00
Dilshod Tadjibaev e882d41f8b
Use updated serde_rusqlite version (MIT/Apache2 license) (#956) 
fixes #719
 2023-11-14 19:03:07 -05:00
chenkun 2614944afa
fix approximately equal precision issue in test code (#954) 2023-11-13 15:35:24 -05:00
Louis Fortier-Dubois 4d63a24e19
fix clippy (#955) 2023-11-13 15:30:07 -05:00
Ariel Cohen cb4c23b39d
add gather (#947) 2023-11-13 08:12:00 -05:00
Louis Fortier-Dubois 831335ac2e
Perf/wgpu/reduce dim (#943) 
* new reduce half working

* surprisingly working

* good on elongated matrix, bad on balanced ones

* working and clean

* autotune not tested, tests fail at non contiguous

* fixed

* autotune tested

* mean dim

* some fixes

* clippy
 2023-11-13 07:20:50 -05:00
Dilshod Tadjibaev f53ab06efc
Pin candle-core version to "0.3.0" version (#950) 
Candle core 0.3.1 release contains a breaking changes so this is a workaround to pin to "0.3.0".
 2023-11-12 17:56:30 -05:00
dengelt e255472826
Fix typos (#949) 2023-11-12 15:08:39 -05:00
Nathaniel Simard 322480b744
Feat/op fusion decorator (#939) 
* WIP

* Impl backend decorator

* WIP

* WIP

* WIP

* WIP

* WIP

* WIP

* Refactor

* Handle graph single ops execution

* WIP

* Starting to get concrete

* WIP

* Fix locator

* Implement add ops

* Start implementing ops

* Add more ops

* Add more ops

* More float ops

* Almost finish float ops

* Almost done with Int

* Some fix

* Into float

* Implement bool ops

* Almost done with MVP

* Fix adaptive pooling

* Add fusion as backend

* Fix memory leak

* Fix

* WIP Doc

* Doc all ops enum

* Initial docs

* Clippy

* Clippy v2

* Fix typos

* Fix doc

* Fix feature flags

* Add missing ops

* Some cleanup

* Revert u128 id

* cosmetic fixes

---------

Co-authored-by: louisfd <louisfd94@gmail.com>
 2023-11-09 21:21:41 -05:00
Louis Fortier-Dubois 6011ae01fd
Update burn-compute README.md to check autotune 2023-11-07 12:26:42 -05:00
Nathaniel Simard c4bc96e27f
Better settings (#933) 2023-11-07 07:34:39 -05:00
Louis Fortier-Dubois a0297530ea
Autotune: fix inputs (#926) 2023-11-06 08:59:31 -05:00
Louis Fortier-Dubois 6548f1a730
add needed lines (#927) 2023-11-03 09:55:33 -04:00
Aisuko 4e8b573f13
Fixed the wrong order of the attributes (#930) 
Signed-off-by: GitHub <noreply@github.com>
 2023-11-03 09:21:58 -04:00
Nathaniel Simard dddc138757
Add warmup logic when calculating eta (#923) 2023-11-03 08:57:09 -04:00
Louis Fortier-Dubois 2ac348c604
fix singular in estimated time (#928) 2023-11-03 08:52:48 -04:00
Louis Fortier-Dubois 1cc1844d32
Refactor/autotune/key (#924) 2023-11-03 08:46:25 -04:00
Luni-4 8c80c9b94a
ci/Speed up typos checks (#907) 2023-11-02 14:30:07 -04:00
Louis Fortier-Dubois 35df31f700
Perf/wgpu/matmul unpadded (#922) 2023-11-01 16:37:33 -04:00
Nathaniel Simard 64e58b4463
Make ndarray tensor public (#920) 2023-11-01 13:31:21 -04:00
Louis Fortier-Dubois 8742d31d16
Perf/wgpu/matmul vec4rhs (#914) 2023-10-31 08:37:17 -04:00
Nathaniel Simard 96524d40a1
[Breaking] Refactor Backend Names (#904) 2023-10-29 18:27:49 -04:00
Louis Fortier-Dubois e2a3329997
Feat/wgpu/autotune compute (#906) 2023-10-29 16:44:59 -04:00
Arvid Hammarlund a9567ab252
Fixing Docs.rs (#905) 2023-10-26 15:26:11 -04:00
Luni-4 7332ebcabf
ci/Add coverage as xtask task (#902) 2023-10-26 12:45:08 -04:00
nathaniel c0f836a94d Update docs link 2023-10-25 11:09:06 -04:00
Louis Fortier-Dubois 068b460078
add bump command (#901) 2023-10-25 08:59:46 -04:00
Nathaniel Simard 233922d60c
Chore: Bump version for next release (#900) 2023-10-24 19:31:13 -04:00
nathaniel 4eb69735e4 Fix publish workflow 2023-10-24 19:26:21 -04:00
louisfd d258778272 candle link 2023-10-24 18:28:10 -04:00
louisfd 68e41d744f Merge branch 'main' of github.com:burn-rs/burn 2023-10-24 18:22:48 -04:00
louisfd aaae336945 candle readme 2023-10-24 18:22:37 -04:00
nathaniel cfb3157e04 Fix publish workflow 2023-10-24 18:08:19 -04:00
Luni-4 9f4eec7fe5
ci: Do not consider `examples` folder for coverage (#898) 2023-10-24 17:25:04 -04:00
Louis Fortier-Dubois e76b6d47de
WGPU: matmul vec4 (#897) 2023-10-24 17:23:43 -04:00
Louis Fortier-Dubois 0ab611b42e
AdamW NaN fix (#888) 2023-10-24 14:48:40 -04:00
Nathaniel Simard 1fd59552db
[Burn-Tensor] Add clone invariance (#891) 
* [Burn-Tensor] Add clone invariance

* Fix div by zero
 2023-10-24 14:45:56 -04:00
nathaniel ae0de594fd CI: Update publish step 2023-10-24 14:37:27 -04:00
Alex Errant 9f2bc599b8
Add a `sync` feature to common, core, and tensor (#893) 2023-10-24 14:32:01 -04:00
nathaniel d021c7d7e8 Remove wrong comments 2023-10-24 11:55:39 -04:00
Luni-4 aa1f3e3f92
ci/Add filters (#892) 2023-10-24 11:12:33 -04:00
Luni-4 9add42442f
Generalize model usage in burn-import README (#889) 2023-10-24 09:53:42 -04:00
Nathaniel Simard 84df5554b1
Use const seed (#894) 2023-10-24 09:53:11 -04:00
Luni-4 38e88a79bd
ci: Implement source-code coverage (#890) 2023-10-23 14:15:14 -04:00
Louis Fortier-Dubois d96f73da0a
Feat/compute/autotune (#861) 
* wip autotune compute

* too much generics

* wip

* megawip

* in progress

* first test passes

* first test passes

* fixed test

* refactor for cache hit and miss

* cleanup and fixes

* doc and stuff

* doc and stuff

* clippy

* format

* remove lifetime

* cleanup operation

* wip

* wip

* compiles

* wip mutable borrow

* refactor with autotune server

* wip tune benchmark

* test passes

* fix autotune key

* cache hit miss tests

* refactor wgpu to match burn-compute

* better operation execution

* cleanup & refactor

* test for parametered kernel

* fmt

* fmt

* clippy

* allow clippy

* fix no-std

* fmt

* review and ci

* Fix CI

* delete dummy benchmarks again

---------

Co-authored-by: nathaniel <nathaniel.simard.42@gmail.com>
 2023-10-23 11:29:44 -04:00
Nathaniel Simard 86db5dc392
Enable candle cuda (#887) 2023-10-23 11:00:54 -04:00
Nathaniel Simard 80fe58c604
[Burn-train] Improve panic messages (#885) 
* [Burn-train] Improve panic messages

* Add new to in-memory logger
 2023-10-23 10:49:46 -04:00
Louis Fortier-Dubois e4d9d67526
make candle available (#886) 2023-10-23 10:00:39 -04:00
Mathias Insley 07c0cf146d
Wgpu/Clamp Kernels (#866) 
* Update kernel mod.rs

* Wgpu crate implementations and add shader files

* Direct backends to the correct implementation

* Use mask method for candle

* Add index out of bounds protection

* Use a macro to avoid duplication

* Use unary_scalar templates

* New shaders for clamp and clamp_inplace

* Remove unneccessary clamp shaders

* Clamp implementation and test

* Use new clamp implementation for float and int ops

* Better variable names for clamp_min/max

* Revert changes to tensor/ops/tensor.rs

* Fix clamp.wgsl

* Fix shader types

* Use native candle clamp

* Use candle ops for clamp_min/max and revert tensor.rs

* Maximum/minimum were reversed
 2023-10-23 07:49:24 -04:00
Nathaniel Simard d263968236
Refactor unfold4d + Add Module (#870) 2023-10-22 11:53:59 -04:00