55d4e0d886
minor cleanup
2024-06-27 12:56:05 -04:00
1746d41358
refactor write for comptime n checks
2024-06-27 11:32:01 -04:00
6b58882b40
refactor for horizontal check comptime
2024-06-27 10:00:22 -04:00
5d9646bb1b
works everytime
2024-06-27 07:57:58 -04:00
6250fa89c8
Remove closed 'future improvements' ( #1935 )
...
Closed issues that are still under the 'future improvements' heading, and have been implemented, have been removed from the heading.
2024-06-27 07:50:28 -04:00
cdd1fa1672
Refactor tensor data ( #1916 )
...
* Move distribution to module
* Add new TensorData with serialization support
* Implement display and from for TensorData
* Add missing Cargo.lock
* Add missing bytemuck feature
* Add zeros, ones, full and random TensorData methods
* Refactor Data -> TensorData usage
* Fix tests
Since TensorData is not generic over the element type anymore no type inference can be done by the compiler. We must explicitly cast the expected results to the expected backend type.
* Remove commented line
* Fix import
* Add record-backward-compat
* Remove dim const generic from TensorData
* Support NestedValue de/serialization with TensorData
* Fix burn-jit tests
* Remove eprinln
* Refactor onnx import to use TensorData
* Fix tch from_data
* Fix nested value serialization for u8
* Fix missing import
* Fix reduce min onnx test
* Fix deprecated attribute
* Remove shape getter
* Remove strict assert in tests
* Add tensor data as_bytes
* Add tensor check for rank mismatch
* Fix typo (dimensions plural)
* Fix error message
* Update book examples with from_data and fix Display impl for TensorData
* Add deprecation note
2024-06-26 20:22:19 -04:00
4cb4f22599
more wip
2024-06-26 18:33:20 -04:00
b93e984555
wip fixing last bug
2024-06-26 17:32:53 -04:00
1c7780aaac
Feat/dynamic small pool ( #1931 )
2024-06-26 15:42:04 -04:00
122b909e75
variable vectorization for output
2024-06-26 15:16:30 -04:00
15483bed4f
works
2024-06-26 14:16:24 -04:00
f9ec2e1006
Handle visibility in cube ( #1929 )
2024-06-26 12:57:47 -04:00
d772a1cfd5
Fix: launch without generics ( #1932 )
2024-06-26 12:57:32 -04:00
d9b4801448
some debugging done
2024-06-26 12:17:27 -04:00
4c9097030f
Perf/dynamic mm slice adressing ( #1917 )
...
* basic implementation of virtual memory addressing for fast index + merging (there is a bug with slice padding
2024-06-25 18:16:46 -04:00
379d15918c
refactoring
2024-06-25 14:45:55 -04:00
2fbc4628f3
Feat/cube/array assign ops ( #1914 )
2024-06-25 09:55:55 -04:00
0f8dd57d9c
Combined PRs ( #1928 )
...
* Bump syn from 2.0.66 to 2.0.68
Bumps [syn](https://github.com/dtolnay/syn ) from 2.0.66 to 2.0.68.
- [Release notes](https://github.com/dtolnay/syn/releases )
- [Commits](https://github.com/dtolnay/syn/compare/2.0.66...2.0.68 )
---
updated-dependencies:
- dependency-name: syn
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump proc-macro2 from 1.0.85 to 1.0.86
Bumps [proc-macro2](https://github.com/dtolnay/proc-macro2 ) from 1.0.85 to 1.0.86.
- [Release notes](https://github.com/dtolnay/proc-macro2/releases )
- [Commits](https://github.com/dtolnay/proc-macro2/compare/1.0.85...1.0.86 )
---
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 wgpu from 0.20.0 to 0.20.1
Bumps [wgpu](https://github.com/gfx-rs/wgpu ) from 0.20.0 to 0.20.1.
- [Release notes](https://github.com/gfx-rs/wgpu/releases )
- [Changelog](https://github.com/gfx-rs/wgpu/blob/v0.20.1/CHANGELOG.md )
- [Commits](https://github.com/gfx-rs/wgpu/compare/v0.20.0...v0.20.1 )
---
updated-dependencies:
- dependency-name: wgpu
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump strum from 0.26.2 to 0.26.3
Bumps [strum](https://github.com/Peternator7/strum ) from 0.26.2 to 0.26.3.
- [Release notes](https://github.com/Peternator7/strum/releases )
- [Changelog](https://github.com/Peternator7/strum/blob/master/CHANGELOG.md )
- [Commits](https://github.com/Peternator7/strum/compare/v0.26.2...v0.26.3 )
---
updated-dependencies:
- dependency-name: strum
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump bytemuck from 1.16.0 to 1.16.1
Bumps [bytemuck](https://github.com/Lokathor/bytemuck ) from 1.16.0 to 1.16.1.
- [Changelog](https://github.com/Lokathor/bytemuck/blob/main/changelog.md )
- [Commits](https://github.com/Lokathor/bytemuck/compare/v1.16.0...v1.16.1 )
---
updated-dependencies:
- dependency-name: bytemuck
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-25 09:23:36 -04:00
2c51615471
Print model structure like with PyTorch - Part 1 ( #1912 )
2024-06-25 09:23:10 -04:00
3faf544bc4
Book: Fix the link to burn-train in "Learner" page ( #1920 )
...
Add the missing "crates/" to the link.
Co-authored-by: towerpark <t56ouhw1d@mozmail.com>
2024-06-25 09:15:34 -04:00
17a01d0131
refactoring
2024-06-22 13:25:59 -04:00
55c117874a
tested write output
2024-06-21 14:48:31 -04:00
418b14f4e0
Update candle-core to a released version ( #1913 )
2024-06-21 08:55:27 -04:00
1404327e70
read tile tests
2024-06-21 08:40:00 -04:00
40a874cdcd
merge main
2024-06-20 16:28:10 -04:00
673f1468b7
wip
2024-06-20 16:11:16 -04:00
a5dfb87828
Feat/comptime expr ( #1910 )
...
* Support comptime expressions
* Add test
* Cleanup
* Fix
2024-06-20 16:00:22 -04:00
c938374ef0
fix
2024-06-20 14:55:27 -04:00
7e3a28a126
tests for computation
2024-06-20 14:52:08 -04:00
fc8d3fb087
merge main
2024-06-20 07:31:03 -04:00
e7161440d9
wip
2024-06-20 07:05:04 -04:00
efc13d9a38
Feat/cube/compile error ( #1909 )
2024-06-19 17:21:32 -04:00
d50bac165e
feat cube support Array ( #1907 )
2024-06-19 17:03:02 -04:00
14d1bbba64
Do not use default burn-compute features unless enabled. ( #1908 )
2024-06-19 10:12:11 -04:00
e651e48099
to vectorized not vectorized
2024-06-19 09:50:28 -04:00
cebd4d4284
debugging wip
2024-06-18 21:48:53 -04:00
560d77d154
Doc: Improve module to_device/fork docs ( #1901 )
2024-06-18 16:45:38 -04:00
e758fd43db
Fix: constant record loading ( #1902 )
2024-06-18 16:45:21 -04:00
263add23a0
Tanh nn wrapper ( #1903 )
2024-06-18 16:45:04 -04:00
f8a7c54272
feat: Make RetroForward public ( #1905 )
2024-06-18 16:44:32 -04:00
96468fc3c9
feat: added reduce min onnx import ( #1894 )
2024-06-18 09:04:24 -04:00
bbd35fd457
Bump clap from 4.5.6 to 4.5.7 ( #1896 )
...
Bumps [clap](https://github.com/clap-rs/clap ) from 4.5.6 to 4.5.7.
- [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/v4.5.6...v4.5.7 )
---
updated-dependencies:
- dependency-name: clap
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com>
2024-06-18 08:57:41 -04:00
4f6db974a1
Perf/dynamic mm ( #1906 )
2024-06-18 08:41:07 -04:00
53cb89bdcb
mega wip
2024-06-17 16:24:54 -04:00
12843ed159
runs, but oh lord is it wrong
2024-06-17 15:00:17 -04:00
43cf06da18
load shared memory with bound check
2024-06-17 12:19:34 -04:00
71d3e258f7
array to_vectorized
2024-06-17 11:36:49 -04:00
8071b637b8
Fix conv2d_weight_grad_groups ( #1891 )
2024-06-17 09:24:33 -04:00
a04da9a285
Combined PRs ( #1900 )
...
* Bump cudarc from 0.11.4 to 0.11.6
Bumps [cudarc](https://github.com/coreylowman/cudarc ) from 0.11.4 to 0.11.6.
- [Release notes](https://github.com/coreylowman/cudarc/releases )
- [Commits](https://github.com/coreylowman/cudarc/compare/v0.11.4...v0.11.6 )
---
updated-dependencies:
- dependency-name: cudarc
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump derive_more from 0.99.17 to 0.99.18
Bumps [derive_more](https://github.com/JelteF/derive_more ) from 0.99.17 to 0.99.18.
- [Release notes](https://github.com/JelteF/derive_more/releases )
- [Changelog](https://github.com/JelteF/derive_more/blob/v0.99.18/CHANGELOG.md )
- [Commits](https://github.com/JelteF/derive_more/compare/v0.99.17...v0.99.18 )
---
updated-dependencies:
- dependency-name: derive_more
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump tokio from 1.37.0 to 1.38.0
Bumps [tokio](https://github.com/tokio-rs/tokio ) from 1.37.0 to 1.38.0.
- [Release notes](https://github.com/tokio-rs/tokio/releases )
- [Commits](https://github.com/tokio-rs/tokio/compare/tokio-1.37.0...tokio-1.38.0 )
---
updated-dependencies:
- dependency-name: tokio
dependency-type: direct:production
update-type: version-update:semver-minor
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump regex from 1.10.4 to 1.10.5
Bumps [regex](https://github.com/rust-lang/regex ) from 1.10.4 to 1.10.5.
- [Release notes](https://github.com/rust-lang/regex/releases )
- [Changelog](https://github.com/rust-lang/regex/blob/master/CHANGELOG.md )
- [Commits](https://github.com/rust-lang/regex/compare/1.10.4...1.10.5 )
---
updated-dependencies:
- dependency-name: regex
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-17 09:06:01 -04:00
ac9f942a46
Remove GraphicsAPI generic for WgpuRuntime ( #1888 )
2024-06-17 09:04:25 -04:00
louisfd