mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-26 10:56:07 +00:00
e66f2beba8
These helpers should not be part of the IR build unit. The interface is now implemented on `builtin.unrealized_conversion_cast` with an external model. Also rename the CastOpInterfaces Bazel target name to CastInterfaces to be consistent with the CMake target name. Differential Revision: https://reviews.llvm.org/D146972
93 lines
3.3 KiB
C++
93 lines
3.3 KiB
C++
//===- CastInterfaces.cpp -------------------------------------------------===//
|
|
//
|
|
// 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/CastInterfaces.h"
|
|
|
|
#include "mlir/IR/BuiltinDialect.h"
|
|
#include "mlir/IR/BuiltinOps.h"
|
|
|
|
using namespace mlir;
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Helper functions for CastOpInterface
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
/// Attempt to fold the given cast operation.
|
|
LogicalResult
|
|
impl::foldCastInterfaceOp(Operation *op, ArrayRef<Attribute> attrOperands,
|
|
SmallVectorImpl<OpFoldResult> &foldResults) {
|
|
OperandRange operands = op->getOperands();
|
|
if (operands.empty())
|
|
return failure();
|
|
ResultRange results = op->getResults();
|
|
|
|
// Check for the case where the input and output types match 1-1.
|
|
if (operands.getTypes() == results.getTypes()) {
|
|
foldResults.append(operands.begin(), operands.end());
|
|
return success();
|
|
}
|
|
|
|
return failure();
|
|
}
|
|
|
|
/// Attempt to verify the given cast operation.
|
|
LogicalResult impl::verifyCastInterfaceOp(Operation *op) {
|
|
auto resultTypes = op->getResultTypes();
|
|
if (resultTypes.empty())
|
|
return op->emitOpError()
|
|
<< "expected at least one result for cast operation";
|
|
|
|
auto operandTypes = op->getOperandTypes();
|
|
if (!cast<CastOpInterface>(op).areCastCompatible(operandTypes, resultTypes)) {
|
|
InFlightDiagnostic diag = op->emitOpError("operand type");
|
|
if (operandTypes.empty())
|
|
diag << "s []";
|
|
else if (llvm::size(operandTypes) == 1)
|
|
diag << " " << *operandTypes.begin();
|
|
else
|
|
diag << "s " << operandTypes;
|
|
return diag << " and result type" << (resultTypes.size() == 1 ? " " : "s ")
|
|
<< resultTypes << " are cast incompatible";
|
|
}
|
|
|
|
return success();
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// External model for BuiltinDialect ops
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
namespace mlir {
|
|
namespace {
|
|
// This interface cannot be implemented directly on the op because the IR build
|
|
// unit cannot depend on the Interfaces build unit.
|
|
struct UnrealizedConversionCastOpInterface
|
|
: CastOpInterface::ExternalModel<UnrealizedConversionCastOpInterface,
|
|
UnrealizedConversionCastOp> {
|
|
static bool areCastCompatible(TypeRange inputs, TypeRange outputs) {
|
|
// `UnrealizedConversionCastOp` is agnostic of the input/output types.
|
|
return true;
|
|
}
|
|
};
|
|
} // namespace
|
|
} // namespace mlir
|
|
|
|
void mlir::builtin::registerCastOpInterfaceExternalModels(
|
|
DialectRegistry ®istry) {
|
|
registry.addExtension(+[](MLIRContext *ctx, BuiltinDialect *dialect) {
|
|
UnrealizedConversionCastOp::attachInterface<
|
|
UnrealizedConversionCastOpInterface>(*ctx);
|
|
});
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// Table-generated class definitions
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Interfaces/CastInterfaces.cpp.inc"
|