Matmul rework
The behavior of the reworked MatMul on the arguments in the following way.
- If both arguments are 2-D they are multiplied like conventional matrices.
- If either argument is N-D, N > 2, it is treated as a stack of matrices residing in the last two indexes.
Exp: For the MatMul of tensor A of shape (3,3,2) with tensor B of shape (3,2,4), A is a stack of three 2D matrices each of shape (3,2), and B is a stack of 3 2D matrices, each of shape (2,4).
Merge request reports
Activity
added 57 commits
-
5f570613...8521b218 - 53 commits from branch
eclipse/aidge:dev
- c450869a - Merge branch 'dev' into matmul_rework
- e2a71d4f - Rework MatMul kernel
- ca472784 - [Upd] MatMul kernel test to handle more cases and add random matrices multiplications tests
- 6b3f1254 - Merge branch 'matmul_rework' of gitlab.eclipse.org:hrouis/aidge_backend_cpu into matmul_rework
Toggle commit list-
5f570613...8521b218 - 53 commits from branch
assigned to @pineapple and @hrouis
requested review from @pineapple
added AddFeature label
changed milestone to %aidge_backend_cpu - v0.2.0
added LanguageC++ StatusReview Ready labels
added 19 commits
-
688e2eaf...c2eb9b34 - 13 commits from branch
eclipse/aidge:dev
- fc36e10e - remove matmul attrs and update kernel
- f70ba8c1 - fix matmul to support more matrix shapes
- 2f459838 - Rework MatMul kernel
- 412c2859 - [Upd] MatMul kernel test to handle more cases and add random matrices multiplications tests
- 1772655e - Improve names of TensorImpl tests
- aa66d4ce - Merge branch 'matmul_rework' of gitlab.eclipse.org:hrouis/aidge_backend_cpu into matmul_rework
Toggle commit list-
688e2eaf...c2eb9b34 - 13 commits from branch
enabled an automatic merge when the pipeline for 34e58b5c succeeds
mentioned in commit 2be78815
mentioned in issue aidge_core#65 (closed)