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

302 Commits

Author SHA1 Message Date
louisfd e87c553c9a benchmark 2024-07-04 14:16:00 -04:00
louisfd 43877da1f2 offset check bounds work 2024-07-04 13:42:26 -04:00
louisfd cbb4de7156 refactor load shared memory file 2024-07-04 10:52:54 -04:00
louisfd 74b06fbf55 transposed fails on purpose 2024-07-04 10:25:04 -04:00
louisfd 9d61e680f3 jit layout info refactor 2024-07-04 10:11:49 -04:00
louisfd 1d05a19406 allow collapsible else if in test 2024-07-03 16:21:38 -04:00
louisfd 8e829da1b6 combine comptime and runtime in elsif 2024-07-03 16:18:39 -04:00
louisfd 2292b38778 comptime else if 2024-07-03 15:54:25 -04:00
louisfd 016456b0d9 clippy 2024-07-03 14:08:45 -04:00
louisfd c4cf3f33cf typo 2024-07-03 13:58:23 -04:00
louisfd 60b9b9fc84 clean tests 2024-07-03 13:56:06 -04:00
louisfd f875793f25 ignore clippy 2024-07-03 13:52:58 -04:00
louisfd 657ad60b14 ignore clippy 2024-07-03 13:52:25 -04:00
louisfd 9773cad04b ignore clippy 2024-07-03 13:52:00 -04:00
louisfd 90bfacad79 fmt 2024-07-03 13:47:54 -04:00
louisfd 623e7a22a4 refactoring 2024-07-03 13:46:44 -04:00
louisfd ed86d91e7a several unrolling options 2024-07-03 12:58:48 -04:00
louisfd d6cee6c402 merge main 2024-07-03 12:40:19 -04:00
Nathaniel Simard 679cfd6dfb
Refactor cube launch + support inplace operation (#1961) 2024-07-03 11:58:35 -04:00
mepatrick73 f5be04f44b
Feat/fixed chunk alloc by class (#1960) 2024-07-03 08:06:05 -04:00
Guillaume Charifi d696d74e3d
Implement 3D and transposed 3D convolutions. (#1945) 
* Implement 3D and transposed 3D convolutions.

* Merge changes from onnx-ir #1921 pr

---------

Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
 2024-07-02 17:54:35 -05:00
Joshua Ferguson 25348cf181
Separating ONNX parsing from burn-import (#1921) 
* separating onnx parsing from burn-import

* ran clippy and cargo-fmt

* removed unused deps from onnx-ir

* fixed clippy warnings that were causing run-checks to fail

* removed dead code

* removed unused dependencies from burn-import

* updated contributor-book, updated publish.yml, added readme

* update cargo lock

* formatted md document with prettier, rephrased sentence

* missed the errors with reduce_prod_conversion during merge

* formatted onnx-to-burn-conversion-tool.md, forgot to save
 2024-07-02 15:17:44 -05:00
louisfd b89ef6e392 cleaner 2024-07-02 13:05:43 -04:00
louisfd 173374ef95 can make empty vector 2024-07-02 13:02:18 -04:00
Guillaume Lagrange 755c0708c4
Fix cmma test (#1957) 2024-07-02 12:10:59 -04:00
louisfd 34b87fc5b4 check sm bounds 2024-07-02 11:06:17 -04:00
louisfd 71cc377793 upstream info access 2024-07-02 10:20:06 -04:00
Dirley Jordan 9e6777d6a5
Add ReduceProd ONNX Import (#1955) 
* Preliminary ReduceProd Support

* Add comma to keep formatter happy

* Give test results a 0.001 tolerance to account for floating-point multiplication

* Reformat assersions

* Correctly mark panic conditions in op_configuration
 2024-07-02 09:05:28 -04:00
louisfd 0c0a357746 merge main 2024-07-02 08:37:23 -04:00
Dilshod Tadjibaev 2bb76283ff
Improve pickle (CandleTensor) conversions to NestedValue (#1944) 
* Manually serialize tensor - fixes #1773

* Rename `value` to `bytes`
 2024-07-02 08:34:19 -04:00
Nathaniel Simard 82a883a57d
Feat/cube/fma (#1947) 2024-07-02 08:32:39 -04:00
Nathaniel Simard cb6b5e7183
Feat/cube/cooperative matrix-multiply and accumulate. (#1943) 2024-07-02 08:31:00 -04:00
Nathaniel Simard ad81a997af
Perf: cube reuse shape and strides (#1939) 2024-07-02 08:28:32 -04:00
Arthur Brussee 849c8f453b
Consistent sync/async handling, allow more functions to be async for wasm. (#1936) 2024-07-02 08:25:28 -04:00
Logan B. Nielsen 3a9367de73
remove manual option matching (#1948) 2024-07-01 10:44:10 -04:00
Guillaume Lagrange e753b0c4e7
Fix output tensor dtype (#1938) 2024-07-01 10:27:31 -04:00
Roy Varon a7efc102b9
Replaced `str` with `Path` (#1919) 
* replaced str with Path

* minor change (Path to AsRef<Path>)

* fixed clippy lint
 2024-06-29 18:17:59 -05:00
louisfd 61ca9ff0b6 unroll transpose 2024-06-28 11:08:56 -04:00
Dilshod Tadjibaev 98a58c867d
Print module - implement module display for remaining modules (part2) (#1933) 2024-06-28 08:37:40 -04:00
louisfd 081fd782af fix unroll bug 2024-06-27 16:59:50 -04:00
louisfd 71ea9fb415 benchmarked, it's slower :( 2024-06-27 16:55:04 -04:00
louisfd 3b473597df &mut in matmul 2024-06-27 16:42:21 -04:00
Nathaniel Simard 1ae1c03b2d
Refactor/cube/mutability (#1934) 2024-06-27 16:03:23 -04:00
louisfd 32c2c1b292 fix vectorization bug 2024-06-27 16:01:23 -04:00
louisfd a3c081ac7d wip merge main 2024-06-27 15:39:27 -04:00
louisfd e20a5f0a81 matmul cube tests refactor 2024-06-27 15:09:05 -04:00
louisfd 4e82ad410c cleaner load shared tests 2024-06-27 14:34:18 -04:00
louisfd 55d4e0d886 minor cleanup 2024-06-27 12:56:05 -04:00
louisfd 1746d41358 refactor write for comptime n checks 2024-06-27 11:32:01 -04:00
louisfd 6b58882b40 refactor for horizontal check comptime 2024-06-27 10:00:22 -04:00