replica
/
hanchenye-Polygeist
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix address of for array convention

This commit is contained in:
William S. Moses 2021-07-08 20:11:41 -04:00 committed by William Moses
parent 4be3c5f158
commit 014817c7e0
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mlir-clang/Lib/clang-mlir.cc
View File

@ -2485,11 +2485,20 @@ ValueWithOffsets MLIRScanner::VisitUnaryOperator(clang::UnaryOperator *U) {
Glob.getMLIRType(U->getSubExpr()->getType(), &isArray);
auto mt = sub.val.getType().cast<MemRefType>();
auto shape = std::vector<int64_t>(mt.getShape());
if (!isArray)
mlir::Value res;
if (!isArray) {
shape[0] = -1;
} else {
shape.insert(shape.begin(), -1);
}
auto mt0 = mlir::MemRefType::get(shape, mt.getElementType(),
mt.getAffineMaps(), mt.getMemorySpace());
return ValueWithOffsets(builder.create<memref::CastOp>(loc, sub.val, mt0),
if (!isArray)
res = builder.create<memref::CastOp>(loc, sub.val, mt0);
else
res = builder.create<polygeist::SubIndexOp>(loc, mt0, sub.val, getConstantIndex(-1));
return ValueWithOffsets(res,
/*isReference*/ false);
}
case clang::UnaryOperator::Opcode::UO_Plus: {