MlirOptMain.cpp File Reference

#include "mlir/Tools/mlir-opt/MlirOptMain.h"
#include "mlir/Bytecode/BytecodeWriter.h"
#include "mlir/Debug/CLOptionsSetup.h"
#include "mlir/Debug/Counter.h"
#include "mlir/Dialect/IRDL/IR/IRDL.h"
#include "mlir/Dialect/IRDL/IRDLLoading.h"
#include "mlir/IR/AsmState.h"
#include "mlir/IR/Attributes.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/IR/Diagnostics.h"
#include "mlir/IR/Location.h"
#include "mlir/IR/MLIRContext.h"
#include "mlir/IR/Remarks.h"
#include "mlir/Parser/Parser.h"
#include "mlir/Pass/PassManager.h"
#include "mlir/Pass/PassRegistry.h"
#include "mlir/Remark/RemarkStreamer.h"
#include "mlir/Support/FileUtilities.h"
#include "mlir/Support/Timing.h"
#include "mlir/Support/ToolUtilities.h"
#include "mlir/Tools/ParseUtilities.h"
#include "mlir/Tools/Plugins/DialectPlugin.h"
#include "mlir/Tools/Plugins/PassPlugin.h"
#include "llvm/ADT/StringRef.h"
#include "llvm/Remarks/RemarkFormat.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/Debug.h"
#include "llvm/Support/InitLLVM.h"
#include "llvm/Support/LogicalResult.h"
#include "llvm/Support/ManagedStatic.h"
#include "llvm/Support/Process.h"
#include "llvm/Support/Regex.h"
#include "llvm/Support/SourceMgr.h"
#include "llvm/Support/ThreadPool.h"
#include "llvm/Support/ToolOutputFile.h"

Go to the source code of this file.

Functions
LogicalResult	loadIRDLDialects (StringRef irdlFile, MLIRContext &ctx)
static LogicalResult	doVerifyRoundTrip (Operation *op, const MlirOptMainConfig &config, bool useBytecode)
static LogicalResult	doVerifyRoundTrip (Operation *op, const MlirOptMainConfig &config)
static LogicalResult	performActions (raw_ostream &os, const std::shared_ptr< llvm::SourceMgr > &sourceMgr, MLIRContext *context, const MlirOptMainConfig &config)
	Perform the actions on the input file indicated by the command line flags within the specified context.
static LogicalResult	processBuffer (raw_ostream &os, std::unique_ptr< MemoryBuffer > ownedBuffer, llvm::MemoryBufferRef sourceBuffer, const MlirOptMainConfig &config, DialectRegistry &registry, SourceMgrDiagnosticVerifierHandler *verifyHandler, llvm::ThreadPoolInterface *threadPool)
	Parses the memory buffer.
static LogicalResult	printRegisteredDialects (DialectRegistry &registry)
static LogicalResult	printRegisteredPassesAndReturn ()

Variables
ManagedStatic< MlirOptMainConfigCLOptions >	clOptionsConfig

Function Documentation

doVerifyRoundTrip() [1/2]

LogicalResult doVerifyRoundTrip ( Operation * op, const MlirOptMainConfig & config )

static

Definition at line 487 of file MlirOptMain.cpp.

References doVerifyRoundTrip(), and success().

doVerifyRoundTrip() [2/2]

LogicalResult doVerifyRoundTrip ( Operation * op, const MlirOptMainConfig & config, bool useBytecode )

static

Definition at line 421 of file MlirOptMain.cpp.

References mlir::MLIRContext::allowsUnregisteredDialects(), mlir::MLIRContext::allowUnregisteredDialects(), mlir::MLIRContext::appendDialectRegistry(), mlir::Operation::emitOpError(), mlir::OwningOpRef< OpTy >::get(), mlir::Operation::getContext(), mlir::MLIRContext::getDialectRegistry(), mlir::MlirOptMainConfig::getIrdlFile(), loadIRDLDialects(), mlir::parseSourceString(), mlir::Operation::print(), mlir::MlirOptMainConfig::shouldVerifyOnParsing(), success(), and mlir::writeBytecodeToFile().

Referenced by doVerifyRoundTrip(), and performActions().

loadIRDLDialects()

LogicalResult loadIRDLDialects	(	StringRef	irdlFile,
		MLIRContext &	ctx )

Definition at line 390 of file MlirOptMain.cpp.

References mlir::MLIRContext::appendDialectRegistry(), mlir::emitError(), mlir::OwningOpRef< OpTy >::get(), mlir::DialectRegistry::insert(), mlir::irdl::loadDialects(), mlir::openInputFile(), and mlir::parseSourceFile().

Referenced by doVerifyRoundTrip(), and processBuffer().

performActions()

LogicalResult performActions ( raw_ostream & os, const std::shared_ptr< llvm::SourceMgr > & sourceMgr, MLIRContext * context, const MlirOptMainConfig & config )

static

Perform the actions on the input file indicated by the command line flags within the specified context.

This typically parses the main source file, runs zero or more optimization passes, then prints the output.

Definition at line 501 of file MlirOptMain.cpp.

References mlir::PassReproducerOptions::apply(), mlir::applyDefaultTimingManagerCLOptions(), mlir::applyPassManagerCLOptions(), mlir::PassReproducerOptions::attachResourceParser(), mlir::MlirOptMainConfig::bytecodeProducerToEmit(), mlir::MlirOptMainConfig::bytecodeVersionToEmit(), mlir::MLIRContext::disableMultithreading(), doVerifyRoundTrip(), mlir::emitError(), mlir::MLIRContext::enableMultithreading(), mlir::remark::enableOptimizationRemarks(), mlir::remark::enableOptimizationRemarksWithLLVMStreamer(), mlir::PassManager::enableTiming(), mlir::PassManager::enableVerifier(), mlir::OwningOpRef< OpTy >::get(), mlir::PassManager::getContext(), mlir::OpPassManager::getOpAnchorName(), mlir::OpPassManager::getPasses(), mlir::MLIRContext::getRemarkEngine(), mlir::MlirOptMainConfig::getRemarkFormat(), mlir::MlirOptMainConfig::getRemarkPolicy(), mlir::MlirOptMainConfig::getRemarksAllFilter(), mlir::MlirOptMainConfig::getRemarksAnalyseFilter(), mlir::MlirOptMainConfig::getRemarksFailedFilter(), mlir::MlirOptMainConfig::getRemarksMissedFilter(), mlir::MlirOptMainConfig::getRemarksOutputFile(), mlir::MlirOptMainConfig::getRemarksPassedFilter(), mlir::MlirOptMainConfig::getReproducerFilename(), mlir::TimingManager::getRootScope(), mlir::OpPassManager::Implicit, mlir::MLIRContext::isMultithreadingEnabled(), mlir::makeReproducer(), mlir::TimingScope::nest(), mlir::parseSourceFileForTool(), mlir::REMARK_FORMAT_BITSTREAM, mlir::REMARK_FORMAT_STDOUT, mlir::REMARK_FORMAT_YAML, mlir::REMARK_POLICY_ALL, mlir::REMARK_POLICY_FINAL, mlir::PassManager::run(), mlir::BytecodeWriterConfig::setDesiredBytecodeVersion(), mlir::BytecodeWriterConfig::setElideResourceDataFlag(), mlir::MlirOptMainConfig::setupPassPipeline(), mlir::MlirOptMainConfig::shouldElideResourceDataFromBytecode(), mlir::MlirOptMainConfig::shouldEmitBytecode(), mlir::MlirOptMainConfig::shouldRunReproducer(), mlir::MlirOptMainConfig::shouldUseExplicitModule(), mlir::MlirOptMainConfig::shouldVerifyOnParsing(), mlir::MlirOptMainConfig::shouldVerifyPasses(), mlir::MlirOptMainConfig::shouldVerifyRoundtrip(), mlir::TimingScope::stop(), success(), and mlir::writeBytecodeToFile().

Referenced by processBuffer().

printRegisteredDialects()

LogicalResult printRegisteredDialects ( DialectRegistry & registry )

static

Definition at line 731 of file MlirOptMain.cpp.

References mlir::DialectRegistry::getDialectNames(), and success().

Referenced by mlir::MlirOptMain(), and mlir::MlirOptMain().

printRegisteredPassesAndReturn()

LogicalResult printRegisteredPassesAndReturn ( )

static

Definition at line 738 of file MlirOptMain.cpp.

References mlir::printRegisteredPasses(), and success().

Referenced by mlir::MlirOptMain(), and mlir::MlirOptMain().

processBuffer()

LogicalResult processBuffer ( raw_ostream & os, std::unique_ptr< MemoryBuffer > ownedBuffer, llvm::MemoryBufferRef sourceBuffer, const MlirOptMainConfig & config, DialectRegistry & registry, SourceMgrDiagnosticVerifierHandler * verifyHandler, llvm::ThreadPoolInterface * threadPool )

static

Parses the memory buffer.

If successfully, run a series of passes against it and print the result.

Definition at line 639 of file MlirOptMain.cpp.

References mlir::MLIRContext::allowUnregisteredDialects(), mlir::MLIRContext::DISABLED, mlir::MlirOptMainConfig::getDebugConfig(), mlir::MlirOptMainConfig::getDiagnosticVerbosityLevel(), mlir::MlirOptMainConfig::getIrdlFile(), loadIRDLDialects(), performActions(), mlir::MLIRContext::printOpOnDiagnostic(), mlir::SourceMgrDiagnosticVerifierHandler::registerInContext(), mlir::MLIRContext::setThreadPool(), mlir::MlirOptMainConfig::shouldAllowUnregisteredDialects(), mlir::MlirOptMainConfig::shouldShowNotes(), mlir::MlirOptMainConfig::shouldVerifyDiagnostics(), and success().

Referenced by mlir::MlirOptMain().

Variable Documentation

clOptionsConfig

ManagedStatic<MlirOptMainConfigCLOptions> clOptionsConfig

Definition at line 346 of file MlirOptMain.cpp.