1//===- DebugExtensionOps.cpp - Debug extension for the Transform dialect --===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8
9#include "mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.h"
10
11#include "mlir/Dialect/Transform/IR/TransformTypes.h"
12#include "llvm/Support/InterleavedRange.h"
13
14using namespace mlir;
15
16#define GET_OP_CLASSES
17#include "mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.cpp.inc"
18
19DiagnosedSilenceableFailure
20transform::EmitRemarkAtOp::apply(transform::TransformRewriter &rewriter,
21 transform::TransformResults &results,
22 transform::TransformState &state) {
23 if (isa<TransformHandleTypeInterface>(Val: getAt().getType())) {
24 auto payload = state.getPayloadOps(value: getAt());
25 for (Operation *op : payload)
26 op->emitRemark() << getMessage();
27 return DiagnosedSilenceableFailure::success();
28 }
29
30 assert(isa<transform::TransformValueHandleTypeInterface>(getAt().getType()) &&
31 "unhandled kind of transform type");
32
33 auto describeValue = [](Diagnostic &os, Value value) {
34 os << "value handle points to ";
35 if (auto arg = llvm::dyn_cast<BlockArgument>(Val&: value)) {
36 os << "a block argument #" << arg.getArgNumber() << " in block #"
37 << std::distance(first: arg.getOwner()->getParent()->begin(),
38 last: arg.getOwner()->getIterator())
39 << " in region #" << arg.getOwner()->getParent()->getRegionNumber();
40 } else {
41 os << "an op result #" << llvm::cast<OpResult>(Val&: value).getResultNumber();
42 }
43 };
44
45 for (Value value : state.getPayloadValues(handleValue: getAt())) {
46 InFlightDiagnostic diag = ::emitRemark(loc: value.getLoc()) << getMessage();
47 describeValue(diag.attachNote(), value);
48 }
49
50 return DiagnosedSilenceableFailure::success();
51}
52
53DiagnosedSilenceableFailure
54transform::EmitParamAsRemarkOp::apply(transform::TransformRewriter &rewriter,
55 transform::TransformResults &results,
56 transform::TransformState &state) {
57 std::string str;
58 llvm::raw_string_ostream os(str);
59 if (getMessage())
60 os << *getMessage() << " ";
61 os << llvm::interleaved(R: state.getParams(value: getParam()));
62 if (!getAnchor()) {
63 emitRemark() << str;
64 return DiagnosedSilenceableFailure::success();
65 }
66 for (Operation *payload : state.getPayloadOps(value: getAnchor()))
67 ::mlir::emitRemark(loc: payload->getLoc()) << str;
68 return DiagnosedSilenceableFailure::success();
69}
70

source code of mlir/lib/Dialect/Transform/DebugExtension/DebugExtensionOps.cpp