cfc9ddaafc
This is the second (and final) step of making "destination style" usable without depending on the Linalg dialect. (The first step was D135129.) This change allows us to provide default bufferization implementations for all destination-style ops. It also allows us to simplify `TilingInterface`. (E.g., `getDestinationOperands` can be removed.) Differential Revision: https://reviews.llvm.org/D136179
72 lines
2.8 KiB
C++
72 lines
2.8 KiB
C++
//===- DestinationStyleOpInterface.cpp -- Destination style ops -----------===//
|
|
//
|
|
// 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/Interfaces/DestinationStyleOpInterface.h"
|
|
|
|
using namespace mlir;
|
|
|
|
namespace mlir {
|
|
#include "mlir/Interfaces/DestinationStyleOpInterface.cpp.inc"
|
|
} // namespace mlir
|
|
|
|
OpOperandVector::operator SmallVector<Value>() {
|
|
SmallVector<Value> result;
|
|
result.reserve(this->size());
|
|
llvm::transform(*this, std::back_inserter(result),
|
|
[](OpOperand *opOperand) { return opOperand->get(); });
|
|
return result;
|
|
}
|
|
|
|
LogicalResult detail::verifyDestinationStyleOpInterface(Operation *op) {
|
|
DestinationStyleOpInterface dstStyleOp =
|
|
cast<DestinationStyleOpInterface>(op);
|
|
|
|
SmallVector<OpOperand *> outputBufferOperands, outputTensorOperands;
|
|
for (OpOperand *operand : dstStyleOp.getOutputOperands()) {
|
|
Type type = operand->get().getType();
|
|
if (type.isa<MemRefType>())
|
|
outputBufferOperands.push_back(operand);
|
|
if (type.isa<RankedTensorType>())
|
|
outputTensorOperands.push_back(operand);
|
|
}
|
|
|
|
// Expect at least one output operand.
|
|
int64_t numInputs = dstStyleOp.getNumInputs();
|
|
int64_t numOutputs = dstStyleOp.getNumOutputs();
|
|
if (numOutputs == 0)
|
|
return op->emitOpError("expected at least one output operand");
|
|
if (failed(OpTrait::impl::verifyNOperands(op, numInputs + numOutputs)))
|
|
return failure();
|
|
// Verify the number of results matches the number of output tensors.
|
|
if (op->getNumResults() != outputTensorOperands.size())
|
|
return op->emitOpError("expected the number of results (")
|
|
<< op->getNumResults()
|
|
<< ") to be equal to the number of output tensors ("
|
|
<< outputTensorOperands.size() << ")";
|
|
|
|
// Simplifying assumption: either full tensor or full buffer mode.
|
|
// This allows simpler verification of output operands vs result types
|
|
// without premature tracking of which operand is what in mixed-mode.
|
|
// TODO: relax when mixed-mode needs to pass verification.
|
|
if (!outputBufferOperands.empty() && !outputTensorOperands.empty())
|
|
return op->emitOpError(
|
|
"expected output operands to all have tensor type or "
|
|
"all have buffer type");
|
|
|
|
for (OpOperand *opOperand : outputTensorOperands) {
|
|
OpResult result = dstStyleOp.getTiedOpResult(opOperand);
|
|
if (result.getType() != opOperand->get().getType())
|
|
return op->emitOpError("expected type of operand #")
|
|
<< opOperand->getOperandNumber() << " ("
|
|
<< opOperand->get().getType() << ")"
|
|
<< " to match type of corresponding result (" << result.getType()
|
|
<< ")";
|
|
}
|
|
return success();
|
|
}
|