93 lines · cpp
1//===- CastInterfaces.cpp -------------------------------------------------===//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/Interfaces/CastInterfaces.h"10 11#include "mlir/IR/BuiltinDialect.h"12#include "mlir/IR/BuiltinOps.h"13 14using namespace mlir;15 16//===----------------------------------------------------------------------===//17// Helper functions for CastOpInterface18//===----------------------------------------------------------------------===//19 20/// Attempt to fold the given cast operation.21LogicalResult22impl::foldCastInterfaceOp(Operation *op, ArrayRef<Attribute> attrOperands,23 SmallVectorImpl<OpFoldResult> &foldResults) {24 OperandRange operands = op->getOperands();25 if (operands.empty())26 return failure();27 ResultRange results = op->getResults();28 29 // Check for the case where the input and output types match 1-1.30 if (operands.getTypes() == results.getTypes()) {31 foldResults.append(operands.begin(), operands.end());32 return success();33 }34 35 return failure();36}37 38/// Attempt to verify the given cast operation.39LogicalResult impl::verifyCastInterfaceOp(Operation *op) {40 auto resultTypes = op->getResultTypes();41 if (resultTypes.empty())42 return op->emitOpError()43 << "expected at least one result for cast operation";44 45 auto operandTypes = op->getOperandTypes();46 if (!cast<CastOpInterface>(op).areCastCompatible(operandTypes, resultTypes)) {47 InFlightDiagnostic diag = op->emitOpError("operand type");48 if (operandTypes.empty())49 diag << "s []";50 else if (llvm::size(operandTypes) == 1)51 diag << " " << *operandTypes.begin();52 else53 diag << "s " << operandTypes;54 return diag << " and result type" << (resultTypes.size() == 1 ? " " : "s ")55 << resultTypes << " are cast incompatible";56 }57 58 return success();59}60 61//===----------------------------------------------------------------------===//62// External model for BuiltinDialect ops63//===----------------------------------------------------------------------===//64 65namespace mlir {66namespace {67// This interface cannot be implemented directly on the op because the IR build68// unit cannot depend on the Interfaces build unit.69struct UnrealizedConversionCastOpInterface70 : CastOpInterface::ExternalModel<UnrealizedConversionCastOpInterface,71 UnrealizedConversionCastOp> {72 static bool areCastCompatible(TypeRange inputs, TypeRange outputs) {73 // `UnrealizedConversionCastOp` is agnostic of the input/output types.74 return true;75 }76};77} // namespace78} // namespace mlir79 80void mlir::builtin::registerCastOpInterfaceExternalModels(81 DialectRegistry ®istry) {82 registry.addExtension(+[](MLIRContext *ctx, BuiltinDialect *dialect) {83 UnrealizedConversionCastOp::attachInterface<84 UnrealizedConversionCastOpInterface>(*ctx);85 });86}87 88//===----------------------------------------------------------------------===//89// Table-generated class definitions90//===----------------------------------------------------------------------===//91 92#include "mlir/Interfaces/CastInterfaces.cpp.inc"93