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

1016 Commits

Author SHA1 Message Date
Edmondo Porcu b86bc58761
Implementing ONNX support for sqrt (#991) 
* Implementing ONNX support for sqrt

* Formatting, my bad

* Implementing feedback from pull request

* Fixing codegen

* Fixing tests

* Fixing tests
 2023-11-22 16:12:35 -06:00
Louis Fortier-Dubois 17f59057d6
Feat/backend comparison/persistence (#979) 
* setting up

* wip

* persistence works

* cleanup

* clippy

* run checks

* Cleanup

* reverse json order

---------

Co-authored-by: nathaniel <nathaniel.simard.42@gmail.com>
 2023-11-22 11:50:27 -05:00
Nathaniel Simard 630044e96b
Fix/devices api (#990) 2023-11-22 10:24:24 -05:00
Nathaniel Simard 3d6c738776
Refactor/fusion/graph (#988) 2023-11-22 09:55:42 -05:00
Sylvain Benner 682e7bcbea
Fix SQLite tests on Windows (#989) 
* Fix sqlite tests on Windows platform

Drop the connection pool so that the database can be persisted.
 2023-11-22 09:55:00 -05:00
Dilshod Tadjibaev be5bb33788
Update getting-started.md (#987) 
Fixes #828
 2023-11-21 17:40:40 -05:00
Nathaniel Simard cabbaab0c4
Fix/constant tensors (#984) 
* Generalize autodiff tensor

* Can have int const module

* Update example

* Support no-std with burn-import

* Fix typos

* Fix alloc problems

* Revert burn-import changes

* Fix examples

* Support Int and Bool Params

* Fix

* Add comment
 2023-11-21 15:27:28 -06:00
Sylvain Benner 2f079e991b
Fix burn version in the Burn book snippet (#985) 
The book assumes the latest version which is 0.10.0 at the time of this commit.
 2023-11-21 15:59:25 -05:00
Dilshod Tadjibaev 445f41bb7b
Support count_include_pad attr in avg_pool2d ONNX (#978) 
Fixes #636
 2023-11-21 12:21:12 -06:00
Dilshod Tadjibaev cb616ed72c
Fix image classifier build failure by upgrading versions (#967) 2023-11-21 09:28:56 -05:00
David Chavez 88b442015d
feature(norm): Add GroupNorm (#963) 
* Add GroupNorm

* Fix implemenation and add tests

* Address PR comments

* Fix formatting

* Update burn book
 2023-11-21 09:22:56 -05:00
Louis Fortier-Dubois 4711db0e18
bump candle to 0.3.1 and conv_transpose_1d (#977) 2023-11-21 09:13:19 -05:00
Luni-4 cdf54d0b40
ci: Use Ubuntu 22.04 instead of latest version (#974) 2023-11-20 16:07:13 -06:00
Dilshod Tadjibaev 5845790f5c
Disable udeps checking till all unused deps are resolved (#969) 2023-11-20 16:46:26 -05:00
Dilshod Tadjibaev 8c235d6d94
Add a checklist to update book if needed (#972) 2023-11-20 11:02:48 -05:00
David Chavez 49e16b6834
feature(tensor): Add unsqueeze_dim helper (#966) 2023-11-20 10:29:40 -05:00
Rodrigo Pandini 20e9066b57
fix with_file_checkpointer use in training example (#973) 2023-11-20 09:17:29 -05:00
Jean-Gab e5c6044062
doc(burn-tensor): Add examples to slice operation to help noobs like me understand what it does (#880) 
* doc(burn-tensor): Add examples to slice operation to help noobs like me understand what it does

* doc(burn-tensor): Improve slice documentation with more comments and clearer examples

* Fix formatting

* Fix typo.

---------

Co-authored-by: jeangab <jg@nationtech.io>
Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 2023-11-19 23:13:56 -06:00
Luni-4 445603401d
ci/Check dependencies (#895) 2023-11-19 10:35:03 -05:00
Nathaniel Simard 8f1526b9c9
Update readme (#962) 2023-11-17 13:04:41 -05:00
Luni-4 ec9df53d4c
ci/Fix `cargo clippy` action (#942) 2023-11-16 19:35:38 -05:00
Nathaniel Simard 945014b7f1
Add new backend comparison benchmark (#958) 
* Add new benchmark

* Remove bad comment

* Add more gelu
 2023-11-16 08:15:21 -05:00
Zsombor c0859dde59
Implement fusing for recip() (#959) 2023-11-15 17:15:01 -05:00
Nathaniel Simard 24014aca33
WGPU: Support elemwise operation fusion (#948) 2023-11-15 15:13:37 -05:00
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