This is an archive of the discontinued LLVM Phabricator instance.

Differential D146831

[flang][hlfir] get shape from HLFIR expr using hlfir.shape_of
ClosedPublic

Authored by tblah on Mar 24 2023, 10:42 AM.

Details

Reviewers
jeanPerier
clementval
Commits
rG08b09d79b129: [flang][hlfir] get shape from HLFIR expr using hlfir.shape_of
Summary

Depends On: D146830

Diff Detail

Revision Contents

PathSize
flang/
lib/
Optimizer/
Builder/
2 lines

Diff 514201

flang/lib/Optimizer/Builder/HLFIRTools.cpp

Show First 20 Lines • Show All 477 Lines • ▼ Show 20 Linesmlir::Value hlfir::genShape(mlir::Location loc, fir::FirOpBuilder &builder,
if (auto shape = tryRetrievingShapeOrShift(entity)) { if (auto shape = tryRetrievingShapeOrShift(entity)) {
if (shape.getType().isa<fir::ShapeType>()) if (shape.getType().isa<fir::ShapeType>())
return shape; return shape;
if (shape.getType().isa<fir::ShapeShiftType>()) if (shape.getType().isa<fir::ShapeShiftType>())
if (auto s = shape.getDefiningOp<fir::ShapeShiftOp>()) if (auto s = shape.getDefiningOp<fir::ShapeShiftOp>())
return builder.create<fir::ShapeOp>(loc, s.getExtents()); return builder.create<fir::ShapeOp>(loc, s.getExtents());
} }
if (entity.getType().isa<hlfir::ExprType>()) if (entity.getType().isa<hlfir::ExprType>())
TODO(loc, "get shape from HLFIR expr without producer holding the shape"); return builder.create<hlfir::ShapeOfOp>(loc, entity.getBase());
// There is no shape lying around for this entity. Retrieve the extents and // There is no shape lying around for this entity. Retrieve the extents and
// build a new fir.shape. // build a new fir.shape.
return builder.create<fir::ShapeOp>(loc, return builder.create<fir::ShapeOp>(loc,
getVariableExtents(loc, builder, entity)); getVariableExtents(loc, builder, entity));
} }
llvm::SmallVector<mlir::Value> llvm::SmallVector<mlir::Value>
hlfir::getIndexExtents(mlir::Location loc, fir::FirOpBuilder &builder, hlfir::getIndexExtents(mlir::Location loc, fir::FirOpBuilder &builder,
▲ Show 20 Lines • Show All 402 Lines • Show Last 20 Lines