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 39 Branches 19 Tags 101 MiB
Commit Graph

1292 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
Dilshod Tadjibaev 6f2ba34382
Print module part3 - Update book (#1940) 
* Update book example guide

* Update Module book section on module display
 2024-07-01 12:42:17 -05:00
Logan B. Nielsen 3a9367de73
remove manual option matching (#1948) 2024-07-01 10:44:10 -04:00
dependabot[bot] 41c7a5cf4b
Bump uuid from 1.8.0 to 1.9.1 (#1951) 
Bumps [uuid](https://github.com/uuid-rs/uuid) from 1.8.0 to 1.9.1.
- [Release notes](https://github.com/uuid-rs/uuid/releases)
- [Commits](https://github.com/uuid-rs/uuid/compare/1.8.0...1.9.1)

---
updated-dependencies:
- dependency-name: uuid
  dependency-type: direct:production
  update-type: version-update:semver-minor
...

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
 2024-07-01 10:43:37 -04:00
Guillaume Lagrange e753b0c4e7
Fix output tensor dtype (#1938) 2024-07-01 10:27:31 -04:00
github-actions[bot] f15896d597
Combined PRs (#1954) 
* Bump log from 0.4.21 to 0.4.22

Bumps [log](https://github.com/rust-lang/log) from 0.4.21 to 0.4.22.
- [Release notes](https://github.com/rust-lang/log/releases)
- [Changelog](https://github.com/rust-lang/log/blob/master/CHANGELOG.md)
- [Commits](https://github.com/rust-lang/log/compare/0.4.21...0.4.22)

---
updated-dependencies:
- dependency-name: log
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>

* Bump clap from 4.5.7 to 4.5.8

Bumps [clap](https://github.com/clap-rs/clap) from 4.5.7 to 4.5.8.
- [Release notes](https://github.com/clap-rs/clap/releases)
- [Changelog](https://github.com/clap-rs/clap/blob/master/CHANGELOG.md)
- [Commits](https://github.com/clap-rs/clap/compare/clap_complete-v4.5.7...v4.5.8)

---
updated-dependencies:
- dependency-name: clap
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>

* Bump reqwest from 0.12.4 to 0.12.5

Bumps [reqwest](https://github.com/seanmonstar/reqwest) from 0.12.4 to 0.12.5.
- [Release notes](https://github.com/seanmonstar/reqwest/releases)
- [Changelog](https://github.com/seanmonstar/reqwest/blob/master/CHANGELOG.md)
- [Commits](https://github.com/seanmonstar/reqwest/compare/v0.12.4...v0.12.5)

---
updated-dependencies:
- dependency-name: reqwest
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>

* Bump serde_json from 1.0.117 to 1.0.119

Bumps [serde_json](https://github.com/serde-rs/json) from 1.0.117 to 1.0.119.
- [Release notes](https://github.com/serde-rs/json/releases)
- [Commits](https://github.com/serde-rs/json/compare/v1.0.117...v1.0.119)

---
updated-dependencies:
- dependency-name: serde_json
  dependency-type: direct:production
  update-type: version-update:semver-patch
...

Signed-off-by: dependabot[bot] <support@github.com>

---------

Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
 2024-07-01 10:02:29 -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