/// Always print operations in the generic form.
OpPrintingFlags &printGenericOpForm();
+ /// Skip printing regions.
+ OpPrintingFlags &skipRegions();
+
/// Do not verify the operation when using custom operation printers.
OpPrintingFlags &assumeVerified();
/// Return if operations should be printed in the generic form.
bool shouldPrintGenericOpForm() const;
+ /// Return if regions should be skipped.
+ bool shouldSkipRegions() const;
+
/// Return if operation verification should be skipped.
bool shouldAssumeVerified() const;
/// Print operations in the generic form.
bool printGenericOpFormFlag : 1;
+ /// Always skip Regions.
+ bool skipRegionsFlag : 1;
+
/// Skip operation verification.
bool assumeVerifiedFlag : 1;
/// Initialize the printing flags with default supplied by the cl::opts above.
OpPrintingFlags::OpPrintingFlags()
: printDebugInfoFlag(false), printDebugInfoPrettyFormFlag(false),
- printGenericOpFormFlag(false), assumeVerifiedFlag(false),
- printLocalScope(false), printValueUsersFlag(false) {
+ printGenericOpFormFlag(false), skipRegionsFlag(false),
+ assumeVerifiedFlag(false), printLocalScope(false),
+ printValueUsersFlag(false) {
// Initialize based upon command line options, if they are available.
if (!clOptions.isConstructed())
return;
return *this;
}
+/// Always skip Regions.
+OpPrintingFlags &OpPrintingFlags::skipRegions() {
+ skipRegionsFlag = true;
+ return *this;
+}
+
/// Do not verify the operation when using custom operation printers.
OpPrintingFlags &OpPrintingFlags::assumeVerified() {
assumeVerifiedFlag = true;
return printGenericOpFormFlag;
}
+/// Return if Region should be skipped.
+bool OpPrintingFlags::shouldSkipRegions() const { return skipRegionsFlag; }
+
/// Return if operation verification should be skipped.
bool OpPrintingFlags::shouldAssumeVerified() const {
return assumeVerifiedFlag;
/// Print the given operation in the generic form.
void printGenericOp(Operation *op, bool printOpName = true) override {
// Consider nested operations for aliases.
- for (Region ®ion : op->getRegions())
- printRegion(region, /*printEntryBlockArgs=*/true,
- /*printBlockTerminators=*/true);
+ if (!printerFlags.shouldSkipRegions()) {
+ for (Region ®ion : op->getRegions())
+ printRegion(region, /*printEntryBlockArgs=*/true,
+ /*printBlockTerminators=*/true);
+ }
// Visit all the types used in the operation.
for (Type type : op->getOperandTypes())
bool printEmptyBlock = false) override {
if (region.empty())
return;
+ if (printerFlags.shouldSkipRegions()) {
+ os << "{/*skip region*/}";
+ return;
+ }
auto *entryBlock = ®ion.front();
print(entryBlock, printEntryBlockArgs, printBlockTerminators);
// Print regions.
if (op->getNumRegions() != 0) {
os << " (";
- interleaveComma(op->getRegions(), [&](Region ®ion) {
- printRegion(region, /*printEntryBlockArgs=*/true,
- /*printBlockTerminators=*/true, /*printEmptyBlock=*/true);
- });
+ if (!printerFlags.shouldSkipRegions()) {
+ interleaveComma(op->getRegions(), [&](Region ®ion) {
+ printRegion(region, /*printEntryBlockArgs=*/true,
+ /*printBlockTerminators=*/true, /*printEmptyBlock=*/true);
+ });
+ } else {
+ os << "/*skip " << op->getNumRegions() << " regions*/";
+ }
os << ')';
}
void OperationPrinter::printRegion(Region ®ion, bool printEntryBlockArgs,
bool printBlockTerminators,
bool printEmptyBlock) {
+ if (printerFlags.shouldSkipRegions()) {
+ os << "{/*skip region*/}";
+ return;
+ }
os << "{" << newLine;
if (!region.empty()) {
auto restoreDefaultDialect =
os << "Generic Form:\n\n```mlir\n";
- // Temporary drop the regions of this operation so that they don't get
- // printed in the output. This helps keeps the size of the output hover
- // small.
- SmallVector<std::unique_ptr<Region>> regions;
- for (Region ®ion : op.op->getRegions()) {
- regions.emplace_back(std::make_unique<Region>());
- regions.back()->takeBody(region);
- }
-
- op.op->print(
- os, OpPrintingFlags().printGenericOpForm().elideLargeElementsAttrs());
+ op.op->print(os, OpPrintingFlags()
+ .printGenericOpForm()
+ .elideLargeElementsAttrs()
+ .skipRegions());
os << "\n```\n";
- // Move the regions back to the current operation.
- for (Region ®ion : op.op->getRegions())
- region.takeBody(*regions.back());
-
return hover;
}
// CHECK-NEXT: "result": {
// CHECK-NEXT: "contents": {
// CHECK-NEXT: "kind": "markdown",
-// CHECK-NEXT: "value": "\"func.func\" : public @foo\n\nGeneric Form:\n\n```mlir\n\"func.func\"() ({\n}) {function_type = (i1) -> (), sym_name = \"foo\"} : () -> ()\n```\n"
+// CHECK-NEXT: "value": "\"func.func\" : public @foo\n\nGeneric Form:\n\n```mlir\n\"func.func\"() (/*skip 1 regions*/) {function_type = (i1) -> (), sym_name = \"foo\"} : () -> ()\n```\n"
// CHECK-NEXT: },
// CHECK-NEXT: "range": {
// CHECK-NEXT: "end": {
// CHECK-NEXT: "result": {
// CHECK-NEXT: "contents": {
// CHECK-NEXT: "kind": "markdown",
-// CHECK-NEXT: "value": "\"func.func\" : public @foo\n\nGeneric Form:\n\n```mlir\n\"func.func\"() ({\n}) {function_type = (i1) -> (), sym_name = \"foo\"} : () -> ()\n```\n"
+// CHECK-NEXT: "value": "\"func.func\" : public @foo\n\nGeneric Form:\n\n```mlir\n\"func.func\"() (/*skip 1 regions*/) {function_type = (i1) -> (), sym_name = \"foo\"} : () -> ()\n```\n"
// CHECK-NEXT: },
// CHECK-NEXT: "range": {
// CHECK-NEXT: "end": {