nir/spirv: Simplify matrix loads/stores