================
@@ -2450,6 +2450,23 @@ Value *ScalarExprEmitter::VisitInitListExpr(InitListExpr
*E) {
llvm::Value *Init = llvm::Constant::getNullValue(EltTy);
V = Builder.CreateInsertElement(V, Init, Idx, "vecinit");
}
+
+ // Matrix initializer lists are in row-major order but the memory layout for
+ // codegen is determined by the -fmatrix-memory-layout flag (default:
+ // column-major). When the memory layout is column-major, we need to shuffle
+ // the elements from row-major to column-major order.
+ if (const auto *MT = E->getType()->getAs<ConstantMatrixType>()) {
+ if (CGF.getLangOpts().getDefaultMatrixMemoryLayout() !=
+ LangOptions::MatrixMemoryLayout::MatrixRowMajor) {
+ unsigned NumRows = MT->getNumRows();
+ unsigned NumCols = MT->getNumColumns();
+ SmallVector<int, 16> Mask;
+ for (unsigned I = 0, N = NumRows * NumCols; I < N; ++I)
+ Mask.push_back((I % NumRows) * NumCols + (I / NumRows));
----------------
Icohedron wrote:
The MatrixBuilder doesn't have a function to return the integer index of an
element when given row and col.
Perhaps I just implement a new helper function along the lines of
`CreateRowMajorToColumnMajorShuffle` within the MatrixBuilder and then call it
here.
https://github.com/llvm/llvm-project/pull/182904
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
