Improve kleidi-ai matmul register usage#5165

Nicoshev wants to merge 1 commit intopytorch:mainpytorch/FBGEMM:mainfrom

Nicoshev:export-D87656468Nicoshev/FBGEMM:export-D87656468Copy head branch name to clipboard