2798b72ae7
Introduce a new extension for simple print-debugging of the transform dialect scripts. The initial version of this extension consists of two ops that are printing the payload objects associated with transform dialect values. Similar ops were already available in the test extenion and several downstream projects, and were extensively used for testing.
71 lines
2.7 KiB
C++
71 lines
2.7 KiB
C++
//===- DebugExtensionOps.cpp - Debug extension for the Transform dialect --===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.h"
|
|
|
|
#include "mlir/Dialect/Transform/IR/TransformDialect.h"
|
|
#include "mlir/Dialect/Transform/IR/TransformTypes.h"
|
|
#include "mlir/IR/OpImplementation.h"
|
|
|
|
using namespace mlir;
|
|
|
|
#define GET_OP_CLASSES
|
|
#include "mlir/Dialect/Transform/DebugExtension/DebugExtensionOps.cpp.inc"
|
|
|
|
DiagnosedSilenceableFailure
|
|
transform::DebugEmitRemarkAtOp::apply(transform::TransformRewriter &rewriter,
|
|
transform::TransformResults &results,
|
|
transform::TransformState &state) {
|
|
if (getAt().getType().isa<TransformHandleTypeInterface>()) {
|
|
auto payload = state.getPayloadOps(getAt());
|
|
for (Operation *op : payload)
|
|
op->emitRemark() << getMessage();
|
|
return DiagnosedSilenceableFailure::success();
|
|
}
|
|
|
|
assert(
|
|
getAt().getType().isa<transform::TransformValueHandleTypeInterface>() &&
|
|
"unhandled kind of transform type");
|
|
|
|
auto describeValue = [](Diagnostic &os, Value value) {
|
|
os << "value handle points to ";
|
|
if (auto arg = llvm::dyn_cast<BlockArgument>(value)) {
|
|
os << "a block argument #" << arg.getArgNumber() << " in block #"
|
|
<< std::distance(arg.getOwner()->getParent()->begin(),
|
|
arg.getOwner()->getIterator())
|
|
<< " in region #" << arg.getOwner()->getParent()->getRegionNumber();
|
|
} else {
|
|
os << "an op result #" << llvm::cast<OpResult>(value).getResultNumber();
|
|
}
|
|
};
|
|
|
|
for (Value value : state.getPayloadValues(getAt())) {
|
|
InFlightDiagnostic diag = ::emitRemark(value.getLoc()) << getMessage();
|
|
describeValue(diag.attachNote(), value);
|
|
}
|
|
|
|
return DiagnosedSilenceableFailure::success();
|
|
}
|
|
|
|
DiagnosedSilenceableFailure transform::DebugEmitParamAsRemarkOp::apply(
|
|
transform::TransformRewriter &rewriter,
|
|
transform::TransformResults &results, transform::TransformState &state) {
|
|
std::string str;
|
|
llvm::raw_string_ostream os(str);
|
|
if (getMessage())
|
|
os << *getMessage() << " ";
|
|
llvm::interleaveComma(state.getParams(getParam()), os);
|
|
if (!getAnchor()) {
|
|
emitRemark() << os.str();
|
|
return DiagnosedSilenceableFailure::success();
|
|
}
|
|
for (Operation *payload : state.getPayloadOps(getAnchor()))
|
|
::mlir::emitRemark(payload->getLoc()) << os.str();
|
|
return DiagnosedSilenceableFailure::success();
|
|
}
|