0bbc1ed30f
Bug/Remove Squeeze Panic for Multiple Dimensions ( #2035 )
...
* Remove panic for squeeze when more than one axis is specified
* Remove extra Model()
* Change script to squeeze all singleton dimensions
* Revert change since burn requires axes to be specified
* Fix input tensor
* Try updating ONNX files again
* Add script for testing multiple axes along with new ONNX file
* Update squeeze.py comments
* Add squeeze_multiple model to tests
* Fix dim_inference
2024-07-22 12:13:07 -05:00
19cd67a9e2
Migration/cubecl ( #2041 )
2024-07-22 11:08:40 -04:00
0d5025edbb
Refactor tensor quantization for q_* ops ( #2025 )
...
* Move QuantizationScheme to burn-tensor
* Refactor QuantizedTensorPrimitive to include the quantization strategy
* Fix QFloat tensor data display
* Refactor quantization methods to use scheme and qparams (on backend device)
* Fix clippy
* Fix fmt
* Add qtensor primitive tests
2024-07-19 10:39:50 -04:00
3204cbe345
Update cargo.lock
2024-07-18 09:15:49 -04:00
b6784684a1
Bump rust minimal version to 1.79
...
That's because bitstream-io, a dependency of rav1e, started using
a feature only in Rust 1.79.
2024-07-18 09:15:49 -04:00
befe6c1601
Added parameter trust_remote_code to hf dataset call. ( #2013 )
...
* Added parameter trust_remote_code to hf dataset call.
* Removed test modul as it may break causing false negatives.
Set default trust_remote_code to false.
Added an example that highlights the usecase.
2024-07-17 16:40:23 -05:00
9804bf81b2
Adding burn::nn::Sigmoid ( #2031 )
2024-07-17 14:34:44 -04:00
ed8a91d48a
Update slice documentation ( #2024 )
2024-07-16 11:59:02 -05:00
1ed62f36f8
Bump gix-tempfile to fix security audit on gix-fs ( #2022 )
2024-07-16 11:41:11 -04:00
2a5d175e14
Combined PRs ( #2021 )
...
* Bump clap from 4.5.8 to 4.5.9
Bumps [clap](https://github.com/clap-rs/clap ) from 4.5.8 to 4.5.9.
- [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.8...v4.5.9 )
---
updated-dependencies:
- dependency-name: clap
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump serde_json from 1.0.119 to 1.0.120
Bumps [serde_json](https://github.com/serde-rs/json ) from 1.0.119 to 1.0.120.
- [Release notes](https://github.com/serde-rs/json/releases )
- [Commits](https://github.com/serde-rs/json/compare/v1.0.119...v1.0.120 )
---
updated-dependencies:
- dependency-name: serde_json
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump sysinfo from 0.30.12 to 0.30.13
Bumps [sysinfo](https://github.com/GuillaumeGomez/sysinfo ) from 0.30.12 to 0.30.13.
- [Changelog](https://github.com/GuillaumeGomez/sysinfo/blob/master/CHANGELOG.md )
- [Commits](https://github.com/GuillaumeGomez/sysinfo/commits/v0.30.13 )
---
updated-dependencies:
- dependency-name: sysinfo
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump syn from 2.0.69 to 2.0.71
Bumps [syn](https://github.com/dtolnay/syn ) from 2.0.69 to 2.0.71.
- [Release notes](https://github.com/dtolnay/syn/releases )
- [Commits](https://github.com/dtolnay/syn/compare/2.0.69...2.0.71 )
---
updated-dependencies:
- dependency-name: syn
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-15 09:35:36 -04:00
0e77e19635
Remove mention of example in backend section of the book ( #2014 )
2024-07-15 09:34:40 -04:00
7661deb258
Fix image-classsification-web + autotune flag usage ( #2011 )
2024-07-15 09:31:54 -04:00
3afff434bd
Module weight quantization ( #2000 )
...
* Add q_into_data and q_reshape
* Fix tch quantize f16 and q_into_data
* Convert to actual dtype/kind in dequantize
* Add module quantization and q_from_data
* Fix clippy
* Add documentation
* Handle deserialize data conversion
* Fix typo
* Add calibration tests
* Fix clippy precision
* Add QTensorOps require_grad methods to avoid dequantizing
* Add Dequantize mapper docs
* Remove dead code
2024-07-15 08:20:37 -04:00
a4123f6c2e
Cube/doc/readme ( #1904 )
...
---------
Co-authored-by: louisfd <louisfd94@gmail.com>
2024-07-12 10:15:17 -04:00
0a33aa363d
Fix cube docs
2024-07-12 09:25:45 -04:00
19f5ad7be5
Refactor/cube/expand & fix double imports ( #2009 )
...
* Refactored function
* WIP
* Basic stuff done
* Fix traits
* Cleanup
* Cleanup
* Cleanup
2024-07-12 09:18:38 -04:00
35345de62a
Feat/cube/slice ( #2004 )
...
* Refactor Variable types
* Sice
* Implement slice wgsl
* handle lifetime correctly
* Add cuda impl
* Update cmma
* Cleanup
* Fix tests
* Fix slice signature
2024-07-11 11:28:53 -04:00
c30ffcf6ac
Enable optimized handling of bytes ( #2003 )
...
* Enable optimized handling of bytes
* Implement byte buffer de/serialization
* Use serde_bytes w/ alloc (no_std compatible)
2024-07-11 07:48:43 -04:00
69be99b802
Cube: Matmul tiling ( #1994 )
2024-07-09 12:43:13 -04:00
c2b6318fc3
Implement ONNX ConstantOfShape ( #1815 )
...
* Feat: burn-import implement ONNX ConstantOfShape
* Introduce shape type and use in ConstantOfShape and Shape
* Add tests for bool and int tensors for ConstantOfShape
* Fix ONNX test generation
* Undo comment
---------
Co-authored-by: Dilshod Tadjibaev <939125+antimora@users.noreply.github.com>
2024-07-08 15:11:59 -05:00
924e3578ee
Fix CI ( #1993 )
2024-07-08 15:55:05 -04:00
96a375e1d9
Combined PRs ( #1990 )
...
* Bump text_placeholder from 0.5.0 to 0.5.1
Bumps [text_placeholder](https://github.com/bernardoamc/text-placeholder ) from 0.5.0 to 0.5.1.
- [Commits](https://github.com/bernardoamc/text-placeholder/commits )
---
updated-dependencies:
- dependency-name: text_placeholder
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump serde from 1.0.203 to 1.0.204
Bumps [serde](https://github.com/serde-rs/serde ) from 1.0.203 to 1.0.204.
- [Release notes](https://github.com/serde-rs/serde/releases )
- [Commits](https://github.com/serde-rs/serde/compare/v1.0.203...v1.0.204 )
---
updated-dependencies:
- dependency-name: serde
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump trybuild from 1.0.96 to 1.0.97
Bumps [trybuild](https://github.com/dtolnay/trybuild ) from 1.0.96 to 1.0.97.
- [Release notes](https://github.com/dtolnay/trybuild/releases )
- [Commits](https://github.com/dtolnay/trybuild/compare/1.0.96...1.0.97 )
---
updated-dependencies:
- dependency-name: trybuild
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump syn from 2.0.68 to 2.0.69
Bumps [syn](https://github.com/dtolnay/syn ) from 2.0.68 to 2.0.69.
- [Release notes](https://github.com/dtolnay/syn/releases )
- [Commits](https://github.com/dtolnay/syn/compare/2.0.68...2.0.69 )
---
updated-dependencies:
- dependency-name: syn
dependency-type: direct:production
update-type: version-update:semver-patch
...
Signed-off-by: dependabot[bot] <support@github.com>
* Bump cudarc from 0.11.6 to 0.11.7
Bumps [cudarc](https://github.com/coreylowman/cudarc ) from 0.11.6 to 0.11.7.
- [Release notes](https://github.com/coreylowman/cudarc/releases )
- [Commits](https://github.com/coreylowman/cudarc/compare/v0.11.6...v0.11.7 )
---
updated-dependencies:
- dependency-name: cudarc
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-08 15:54:54 -04:00
e7a3cc4fba
Fix wgsl remainder definition ( #1979 )
2024-07-08 15:26:10 -04:00
e8b915a2da
Enhance slice operation to support more range variation ( #1989 )
...
* Enhance slice operation to support more range variation
* Fix doc clippy
* Fixed doc test
* Fix flipped attribute names
* Fix clippy
2024-07-08 13:34:25 -05:00
c0211e2f94
Add static tensor quantization ( #1963 )
...
* Add QuantizationBackend, QTensorOps and QTensor
* Refactor QTensorOps as part of Backend trait
* Add tensor dequantize, QFloat dtype and default affine/symmetric quant
* Add ndarray default quantization implementation
* Fix clippy
* Add rayon parallel iter
* Add quantization operations to book
* Add q_shape and q_device ops to avoid converting the tensor just to get attributes
* Implement autodiff grad ops
* Mark autodiff todo for QAT
* Remove note
* Add q_inner and q_from_inner
2024-07-08 10:16:58 -04:00
6f158af4b1
Fix warnings when using `record-backward-compat` ( #1977 )
2024-07-08 07:58:50 -04:00
8af2b719a1
Feat: Support trait with CubeCL ( #1980 )
2024-07-07 10:07:51 -04:00
c9e9054167
Enable negative starts and ends for slice op ( #1981 )
...
* Enable negative starts and ends for slice op
* Refactor slice_config
---------
Co-authored-by: JC <you@example.com>
2024-07-06 21:31:37 -05:00
3f9e97946f
Feat: Dynamic cube count dispatch ( #1975 )
2024-07-06 19:17:01 -04:00
b331290f8a
Refactor/jit/unary ( #1965 )
2024-07-05 19:47:24 -04:00
d6efb3ca17
Set DEFAULT_MAX_TASKS to 1 when running tests
2024-07-05 18:57:01 -04:00
882a27c52c
Revert "Revert "Implement 3D and transposed 3D convolutions. ( #1945 )""
...
This reverts commit b8b47ea6e6
2024-07-05 18:57:01 -04:00
0928a52eea
Always derive Cube features from adapter ( #1958 )
2024-07-05 17:38:07 -04:00
fe0544b9ea
Add subtract tensor from scalar for ONNX sub op ( #1964 )
2024-07-05 13:52:02 -05:00
1ad2a63f28
Fix typo
2024-07-05 09:40:32 -04:00
b8b47ea6e6
Revert "Implement 3D and transposed 3D convolutions. ( #1945 )"
...
This reverts commit d696d74e3d
2024-07-05 09:40:32 -04:00
51aea94a30
Dynamic memory management preset + updated wgpu buffer memory management ( #1962 )
...
---------
Co-authored-by: mepatrick73 <pameu17@ulaval.ca>
2024-07-04 16:47:08 -04:00
5236e12c81
Add models and examples reference ( #1966 )
...
Co-authored-by: Sylvain Benner <sylvain@benner.online>
---------
Co-authored-by: Sylvain Benner <sylvain@benner.online>
2024-07-04 16:22:08 -04:00
f709858a8b
Revert "Perf: cube reuse shape and strides ( #1939 )" ( #1967 )
...
This reverts commit ad81a997af
2024-07-04 16:16:17 -04:00
679cfd6dfb
Refactor cube launch + support inplace operation ( #1961 )
2024-07-03 11:58:35 -04:00
f5be04f44b
Feat/fixed chunk alloc by class ( #1960 )
2024-07-03 08:06:05 -04:00
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
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
755c0708c4
Fix cmma test ( #1957 )
2024-07-02 12:10:59 -04:00
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
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
82a883a57d
Feat/cube/fma ( #1947 )
2024-07-02 08:32:39 -04:00
cb6b5e7183
Feat/cube/cooperative matrix-multiply and accumulate. ( #1943 )
2024-07-02 08:31:00 -04:00
ad81a997af
Perf: cube reuse shape and strides ( #1939 )
2024-07-02 08:28:32 -04:00
849c8f453b
Consistent sync/async handling, allow more functions to be async for wasm. ( #1936 )
2024-07-02 08:25:28 -04:00
Mathias Insley