change slightly the pzgemm data layout