Skip to content

MLBEDSW-9860 Improve decomposition block selection

Jacob Bohlin requested to merge MLBEDSW-9860 into main

This commit aims to select larger blocks for decomposition and only decompose into block-config blocks when necsessary, i.e. when accumulator load/store functionality is required.

This significantly reduces the number of sub-operators produced when decomposing e.g. large tensor axes and by extension compilation time.

Also did some minor refactoring to utilise OperatorQuery instead of hardcoded conditions for decomposition.

Change-Id: Ida8586a65805f0777718364365639e3dade5742c Signed-off-by: Jacob Bohlin jacob.bohlin@arm.com

Merge request reports