70 lines · cpp
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-exception6//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_CLASSES17#include "mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.cpp.inc"18 19DiagnosedSilenceableFailure20transform::EmitRemarkAtOp::apply(transform::TransformRewriter &rewriter,21 transform::TransformResults &results,22 transform::TransformState &state) {23 if (isa<TransformHandleTypeInterface>(getAt().getType())) {24 auto payload = state.getPayloadOps(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>(value)) {36 os << "a block argument #" << arg.getArgNumber() << " in block #"37 << std::distance(arg.getOwner()->getParent()->begin(),38 arg.getOwner()->getIterator())39 << " in region #" << arg.getOwner()->getParent()->getRegionNumber();40 } else {41 os << "an op result #" << llvm::cast<OpResult>(value).getResultNumber();42 }43 };44 45 for (Value value : state.getPayloadValues(getAt())) {46 InFlightDiagnostic diag = ::emitRemark(value.getLoc()) << getMessage();47 describeValue(diag.attachNote(), value);48 }49 50 return DiagnosedSilenceableFailure::success();51}52 53DiagnosedSilenceableFailure54transform::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(state.getParams(getParam()));62 if (!getAnchor()) {63 emitRemark() << str;64 return DiagnosedSilenceableFailure::success();65 }66 for (Operation *payload : state.getPayloadOps(getAnchor()))67 ::mlir::emitRemark(payload->getLoc()) << str;68 return DiagnosedSilenceableFailure::success();69}70