[Matrix] Add align info to some more loads/stores (NFC).

Some tests were missing alignment info. Subsequent changes properly
preserve the set alignment. Set it properly beforehand, to avoid
unnecessary test changes.