e3cadc4ad3
Rename per-tensor strategies
2024-06-06 13:32:04 -04:00
9f25cec4a6
Data should not directly quantize the values
2024-06-06 13:22:13 -04:00
07b0bdd38f
Add int8 precision settings to book
2024-06-06 12:29:23 -04:00
6b6d6bb676
Add data quantization test
2024-06-06 11:15:01 -04:00
af94ad0a5e
Fix symmetric quantization
2024-06-06 10:27:31 -04:00
96b46b59d0
Fix test_param_serde len
...
New option field ', "quantization": Default(None)' (+31 characters)
2024-06-06 10:00:53 -04:00
26c6b6fea2
Remove skip_serializing_if None (bincode is not self-describing)
...
https://github.com/bincode-org/bincode/issues/326
https://github.com/serde-rs/serde/issues/1732
2024-06-06 09:43:57 -04:00
ad0ace3233
Fix missing import
2024-06-06 09:01:09 -04:00
1fc0197d0d
Fix clippy
2024-06-06 08:15:05 -04:00
ef420cfefa
Implement int8 affine and symmetric quantization
2024-06-05 16:04:04 -04:00
3eb0b3cc21
Add int8 precision settings
2024-06-05 16:03:42 -04:00
36694d5f01
Add data quantization strategy
2024-06-05 16:03:42 -04:00
d28183c7e4
LearnerBuilder "with_checkpointing_strategy" should use builder pattern ( #1841 )
2024-06-05 07:55:44 -04:00
e0a1094f89
Add a feature to initialize from an existing wgpu adapter/device/queue ( #1788 )
...
* Add a feature to initialize from an existing wgpu adapter/device/queue
This is useful when interacting with other wgpu applications (eg. displaying a burn tensor as a texture in egui). The existing devices are keyed by the wgpu Device ID. Alternatively they could be keyed per adapter which would be more inline with other burn WgpuDevice's (one per adapter), but also there's no real inherent reason to.
This also involves making Queue into an Arc. Alternatively, this could give up ownership of the queue, but it's helpful to be able to synchronize burn operations and custom wgpu operations.
2024-06-05 07:19:52 -04:00
36ed65a5cd
Feat/dynamic mm basic implementation + small refactor ( #1844 )
2024-06-04 17:01:33 -04:00
c42abadfe9
Cube: CubeType (no launch) and Comptime::map ( #1853 )
2024-06-04 13:43:43 -04:00
a5af19b959
feat: add sum onnx import ( #1846 )
2024-06-03 15:30:44 -05:00
8a950c667b
Combined PRs ( #1852 )
...
* Bump proc-macro2 from 1.0.84 to 1.0.85
Bumps [proc-macro2](https://github.com/dtolnay/proc-macro2 ) from 1.0.84 to 1.0.85.
- [Release notes](https://github.com/dtolnay/proc-macro2/releases )
- [Commits](https://github.com/dtolnay/proc-macro2/compare/1.0.84...1.0.85 )
---
updated-dependencies:
- dependency-name: proc-macro2
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump syn from 2.0.65 to 2.0.66
Bumps [syn](https://github.com/dtolnay/syn ) from 2.0.65 to 2.0.66.
- [Release notes](https://github.com/dtolnay/syn/releases )
- [Commits](https://github.com/dtolnay/syn/compare/2.0.65...2.0.66 )
---
updated-dependencies:
- dependency-name: syn
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump zip from 2.1.1 to 2.1.2
Bumps [zip](https://github.com/zip-rs/zip2 ) from 2.1.1 to 2.1.2.
- [Release notes](https://github.com/zip-rs/zip2/releases )
- [Changelog](https://github.com/zip-rs/zip2/blob/master/CHANGELOG.md )
- [Commits](https://github.com/zip-rs/zip2/compare/v2.1.1...v2.1.2 )
---
updated-dependencies:
- dependency-name: zip
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump serde from 1.0.202 to 1.0.203
Bumps [serde](https://github.com/serde-rs/serde ) from 1.0.202 to 1.0.203.
- [Release notes](https://github.com/serde-rs/serde/releases )
- [Commits](https://github.com/serde-rs/serde/compare/v1.0.202...v1.0.203 )
---
updated-dependencies:
- dependency-name: serde
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump anyhow from 1.0.83 to 1.0.86
Bumps [anyhow](https://github.com/dtolnay/anyhow ) from 1.0.83 to 1.0.86.
- [Release notes](https://github.com/dtolnay/anyhow/releases )
- [Commits](https://github.com/dtolnay/anyhow/compare/1.0.83...1.0.86 )
---
updated-dependencies:
- dependency-name: anyhow
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-06-03 12:41:02 -04:00
5edaeabcee
Feat/cube/struct support ( #1842 )
...
* struct support (receive, use and modify fields)
* support struct with generics
* expect instead of unwrap
* fmt
* rename struc
* fmt
* Clippy
* Fix launcher
* Support creating private cube type without generics
* Cleanup
* generics support
* clippy
* minor
* fmt
---------
Co-authored-by: nathaniel <nathaniel.simard.42@gmail.com>
2024-06-03 12:19:05 -04:00
92b0067693
Feat/gather import ( #1843 )
...
* Move and redirect GatherElements to new folders/nodes
* Create PyTorch script for gather
* Add onnx file for gather
* Add a gather test to onnx_tests
* Update gather.rs to use select
* Rename codegen test
* Update gather and gather_elements conversion functions
* Validate rank of input node and update output
* Add check for Gather
2024-06-03 08:28:32 -04:00
e407c76def
Remainder operator doc ( #1836 )
...
* Adds remainder ops implementation for Tensor.
* Adds test for % operator.
* Add remainder and % operator entry in tensor.md
---------
Co-authored-by: Jonas Kantic <jk.mail@posteo.net>
2024-06-01 16:49:54 -05:00
fba1e27e0c
Remainder operator ( #1726 )
...
* Adds remainder ops implementation for Tensor.
* Adds test for % operator.
2024-06-01 16:47:02 -05:00
99e1ba4864
feat: expand onnx import ( #1813 )
...
* feat: added expand to import
2024-05-31 16:48:02 -05:00
44f1053219
feat: added range onnx import ( #1834 )
...
* feat: added range onnx import
* fix: range input types
2024-05-31 16:40:54 -05:00
36d4bcd705
[Refactor - Breaking] Refactor cube operations with better names & Support subgroup operations ( #1839 )
2024-05-31 17:07:21 -04:00
13a6f84bc3
Feature/onnx argmax ( #1814 )
...
* pre-test
* implementing argmax for burn-import from onnx
* tidying
* fixing return types and tests
* addressing feedback
* only warn when select_last_index!=0
2024-05-31 14:46:09 -04:00
de0b49e4a3
Cube: Topology constants ( #1838 )
...
---------
Co-authored-by: nathaniel <nathaniel.simard.42@gmail.com>
2024-05-30 12:03:30 -04:00
0d4374cfd4
Update to zip 2.1.1 ( #1837 )
2024-05-29 10:52:39 -04:00
61c9fdbbc8
Cube: cleaner use of topology values ( #1835 )
...
* constant keyword parsing
* works
2024-05-29 09:08:10 -04:00
a2ad424fc8
Indices Operator ( #1735 )
2024-05-29 09:05:31 -04:00
cacc764205
Cube: support for shared memory ( #1831 )
2024-05-29 08:22:04 -04:00
e4836241e1
Fix `DataSerialize` conversion for elements of the same type ( #1832 )
2024-05-28 18:12:44 -04:00
912ffb4a81
Fix wasm examples ( #1824 )
2024-05-28 17:51:20 -04:00
e61b026918
Cube: support method call + prettier tensor metadata ( #1829 )
2024-05-27 15:18:17 -04:00
fd54a8b470
Add vectorization support into cube ( #1830 )
2024-05-27 14:21:29 -04:00
dc85daa1c6
Cube: support for return + conv2d early return ( #1828 )
2024-05-27 13:19:00 -04:00
15d2055de8
Feat/cube/launch ( #1827 )
2024-05-27 12:15:06 -04:00
c59a3b8b8a
Fix bench load record benchmarks ( #1826 )
2024-05-27 10:29:09 -04:00
cccd96de48
Feat: Implement ONNX RandomUniform + RandomNormal in burn-import ( #1806 )
2024-05-27 10:07:04 -04:00
85ba167582
Combined PRs ( #1823 )
...
* Bump cudarc from 0.10.0 to 0.11.0
Bumps [cudarc](https://github.com/coreylowman/cudarc ) from 0.10.0 to 0.11.0.
- [Release notes](https://github.com/coreylowman/cudarc/releases )
- [Commits](https://github.com/coreylowman/cudarc/compare/v0.10.0...v0.11.0 )
---
updated-dependencies:
- dependency-name: cudarc
dependency-type: direct:production
update-type: version-update:semver-minor
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump libc from 0.2.154 to 0.2.155
Bumps [libc](https://github.com/rust-lang/libc ) from 0.2.154 to 0.2.155.
- [Release notes](https://github.com/rust-lang/libc/releases )
- [Commits](https://github.com/rust-lang/libc/compare/0.2.154...0.2.155 )
---
updated-dependencies:
- dependency-name: libc
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump ratatui from 0.26.2 to 0.26.3
Bumps [ratatui](https://github.com/ratatui-org/ratatui ) from 0.26.2 to 0.26.3.
- [Release notes](https://github.com/ratatui-org/ratatui/releases )
- [Changelog](https://github.com/ratatui-org/ratatui/blob/main/CHANGELOG.md )
- [Commits](https://github.com/ratatui-org/ratatui/compare/v0.26.2...v0.26.3 )
---
updated-dependencies:
- dependency-name: ratatui
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump bytemuck from 1.15.0 to 1.16.0
Bumps [bytemuck](https://github.com/Lokathor/bytemuck ) from 1.15.0 to 1.16.0.
- [Changelog](https://github.com/Lokathor/bytemuck/blob/main/changelog.md )
- [Commits](https://github.com/Lokathor/bytemuck/compare/v1.15.0...v1.16.0 )
---
updated-dependencies:
- dependency-name: bytemuck
dependency-type: direct:production
update-type: version-update:semver-minor
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump proc-macro2 from 1.0.83 to 1.0.84
Bumps [proc-macro2](https://github.com/dtolnay/proc-macro2 ) from 1.0.83 to 1.0.84.
- [Release notes](https://github.com/dtolnay/proc-macro2/releases )
- [Commits](https://github.com/dtolnay/proc-macro2/compare/1.0.83...1.0.84 )
---
updated-dependencies:
- dependency-name: proc-macro2
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-05-27 10:04:27 -04:00
1c5e65ab26
docs: update README.md ( #1810 )
...
minor fix
2024-05-27 09:21:03 -04:00
c7ad25ab60
Update cuda-jit ( #1799 )
2024-05-24 11:31:47 -04:00
23c622a9f8
Feat/cube/remaining ops ( #1807 )
2024-05-24 09:48:34 -04:00
1670a71711
Fix burn-jit compile error ( #1803 )
2024-05-23 20:24:42 -04:00
ef4646c90f
feat: Greater + GreaterOrEqual onnx import ( #1801 )
2024-05-23 08:59:15 -04:00
1f31e20ce8
feat: Less + LessOrEqual onnx import ( #1800 )
2024-05-23 08:04:44 -04:00
e39b4d2da0
refactor reduce into separate traits ( #1798 )
2024-05-22 16:01:27 -04:00
033171920c
Cube: first ported kernel + comptime support + variable reuse + cleanup ( #1797 )
2024-05-22 14:08:21 -04:00
b466fd7606
Add seq start position when applying RoPE encoding ( #1796 )
2024-05-22 13:18:31 -04:00
0918cf00c6
feat: added min onnx import ( #1778 )
2024-05-22 10:52:19 -04:00
Guillaume Lagrange