Refactor loop fusion dict / enable matri xmult folding