1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
|
#include "mlir/Remark/RemarkStreamer.h"
#include "mlir/IR/MLIRContext.h"
#include "mlir/IR/Remarks.h"
#include "llvm/Remarks/RemarkSerializer.h"
#include "llvm/Remarks/RemarkStreamer.h"
#include "llvm/Support/Error.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/ToolOutputFile.h"
namespace mlir::remark::detail {
FailureOr<std::unique_ptr<MLIRRemarkStreamerBase>>
LLVMRemarkStreamer::createToFile(llvm::StringRef path,
llvm::remarks::Format fmt) {
std::error_code ec;
// Use error_code ctor; YAML is text. (Bitstream also works fine here.)
auto f =
std::make_unique<llvm::ToolOutputFile>(path, ec, llvm::sys::fs::OF_Text);
if (ec)
return failure();
auto serOr = llvm::remarks::createRemarkSerializer(fmt, f->os());
if (!serOr) {
llvm::consumeError(serOr.takeError());
return failure();
}
auto rs =
std::make_unique<llvm::remarks::RemarkStreamer>(std::move(*serOr), path);
auto impl = std::unique_ptr<LLVMRemarkStreamer>(new LLVMRemarkStreamer());
impl->remarkStreamer = std::move(rs);
impl->file = std::move(f);
return std::unique_ptr<MLIRRemarkStreamerBase>(std::move(impl));
}
void LLVMRemarkStreamer::streamOptimizationRemark(const Remark &remark) {
if (!remarkStreamer->matchesFilter(remark.getCategoryName()))
return;
// First, convert the diagnostic to a remark.
llvm::remarks::Remark r = remark.generateRemark();
// Then, emit the remark through the serializer.
remarkStreamer->getSerializer().emit(r);
}
LLVMRemarkStreamer::~LLVMRemarkStreamer() {
if (file && remarkStreamer)
file->keep();
}
void LLVMRemarkStreamer::finalize() {
if (!remarkStreamer)
return;
remarkStreamer->releaseSerializer();
}
} // namespace mlir::remark::detail
namespace mlir::remark {
LogicalResult enableOptimizationRemarksWithLLVMStreamer(
MLIRContext &ctx, StringRef path, llvm::remarks::Format fmt,
std::unique_ptr<detail::RemarkEmittingPolicyBase> remarkEmittingPolicy,
const RemarkCategories &cat, bool printAsEmitRemarks) {
FailureOr<std::unique_ptr<detail::MLIRRemarkStreamerBase>> sOr =
detail::LLVMRemarkStreamer::createToFile(path, fmt);
if (failed(sOr))
return failure();
return remark::enableOptimizationRemarks(ctx, std::move(*sOr),
std::move(remarkEmittingPolicy), cat,
printAsEmitRemarks);
}
} // namespace mlir::remark
|
