experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Do not emit aliases when printing local form

Browse Source 
Expand local scope printing to skip printing aliases as aliases are printed out at the top of a module and may not be part of the output generated by local scope print.

PiperOrigin-RevId: 280278617
Change-Id: If8bdd2ea204b47cb4e576d025fdfacc9e405e364
This commit is contained in:
Jacques Pienaar 2019-11-13 14:21:16 -08:00 committed by TensorFlower Gardener
parent 0c6be60529
commit d8bba74873
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
third_party/mlir/lib/IR/AsmPrinter.cpp vendored
View File

@ -84,6 +84,11 @@ static llvm::cl::opt<bool>
llvm::cl::desc("Print the generic op form"),
llvm::cl::init(false), llvm::cl::Hidden);
static llvm::cl::opt<bool> printLocalScopeOpt(
"mlir-print-local-scope",
llvm::cl::desc("Print assuming in local scope by default"),
llvm::cl::init(false), llvm::cl::Hidden);
/// Initialize the printing flags with default supplied by the cl::opts above.
OpPrintingFlags::OpPrintingFlags()
: elementsAttrElementLimit(
@ -92,7 +97,8 @@ OpPrintingFlags::OpPrintingFlags()
: Optional<int64_t>()),
printDebugInfoFlag(printDebugInfoOpt),
printDebugInfoPrettyFormFlag(printPrettyDebugInfoOpt),
printGenericOpFormFlag(printGenericOpFormOpt), printLocalScope(false) {}
printGenericOpFormFlag(printGenericOpFormOpt),
printLocalScope(printLocalScopeOpt) {}
/// Enable the elision of large elements attributes, by printing a '...'
/// instead of the element data, when the number of elements is greater than
@ -2020,7 +2026,9 @@ void Block::printAsOperand(raw_ostream &os, bool printType) {
void ModuleOp::print(raw_ostream &os, OpPrintingFlags flags) {
ModuleState state(getContext());
state.initialize(*this);
// Skip initializing in local scope to avoid populating aliases.
if (!flags.shouldUseLocalScope())
state.initialize(*this);
ModulePrinter(os, flags, &state).print(*this);
}