maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Convert MLIR DiagnosticKind to LLVM DiagKind when emitting diagnostic via mlir-opt. 

PiperOrigin-RevId: 222147297
This commit is contained in:
Jacques Pienaar 2018-11-19 15:15:36 -08:00 committed by jpienaar
parent 1c95796372
commit f10f48ee63
1 changed files with 15 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mlir/tools/mlir-opt/mlir-opt.cpp
View File

@ -170,6 +170,18 @@ static StringRef getDiagnosticKindString(MLIRContext::DiagnosticKind kind) {
}
}
/// Given a diagnostic kind, returns the LLVM DiagKind.
static llvm::SourceMgr::DiagKind getDiagKind(MLIRContext::DiagnosticKind kind) {
switch (kind) {
case MLIRContext::DiagnosticKind::Note:
return llvm::SourceMgr::DK_Note;
case MLIRContext::DiagnosticKind::Warning:
return llvm::SourceMgr::DK_Warning;
case MLIRContext::DiagnosticKind::Error:
return llvm::SourceMgr::DK_Error;
}
}
/// Parses the memory buffer. If successfully, run a series of passes against
/// it and print the result.
static OptResult processFile(std::unique_ptr<MemoryBuffer> ownedBuffer) {
@ -189,13 +201,14 @@ static OptResult processFile(std::unique_ptr<MemoryBuffer> ownedBuffer) {
context.registerDiagnosticHandler([&](Location location, StringRef message,
MLIRContext::DiagnosticKind kind) {
unsigned line = 1, column = 1;
SMLoc loc;
if (auto fileLoc = location.dyn_cast<FileLineColLoc>()) {
line = fileLoc->getLine();
column = fileLoc->getColumn();
loc = getLocFromLineAndCol(buffer, line, column);
}
auto unexpectedLoc = getLocFromLineAndCol(buffer, line, column);
sourceMgr.PrintMessage(unexpectedLoc, SourceMgr::DK_Error, message);
sourceMgr.PrintMessage(loc, getDiagKind(kind), message);
});
// Run the test actions.