[MatMul] Add implementation of MatMul export as MatMulCPP
Add export CPP for MatMul operator: - supports scalar/1D inputs - provision for activation as done in elemwise operators - new kernel matmul.hpp provided - TODO: as of now do not support >2D inputs/broadcasts, see below This change does no include the support for >2D inputs. It would require managing multi broadcast which is also not supported for CPP export of elemwise operators. This support is delayed until a further change factorizing multi-broadcast on elemwise/matmul and after availability of a more comprehensive unit tests suite.
Showing
- aidge_export_cpp/kernels/matmul.hpp 33 additions, 0 deletionsaidge_export_cpp/kernels/matmul.hpp
- aidge_export_cpp/operators.py 65 additions, 0 deletionsaidge_export_cpp/operators.py
- aidge_export_cpp/templates/configuration/matmul_config.jinja 15 additions, 0 deletionsaidge_export_cpp/templates/configuration/matmul_config.jinja
- aidge_export_cpp/templates/kernel_forward/matmul_forward.jinja 5 additions, 0 deletions..._export_cpp/templates/kernel_forward/matmul_forward.jinja
Loading