54 lines · c
1//===- Pattern.h - SPIRV Common Conversion Patterns -----------------------===//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#ifndef MLIR_CONVERSION_SPIRVCOMMON_PATTERN_H10#define MLIR_CONVERSION_SPIRVCOMMON_PATTERN_H11 12#include "mlir/Dialect/SPIRV/IR/SPIRVOpTraits.h"13#include "mlir/IR/TypeUtilities.h"14#include "mlir/Transforms/DialectConversion.h"15#include "llvm/Support/FormatVariadic.h"16 17namespace mlir {18namespace spirv {19 20/// Converts elementwise unary, binary and ternary standard operations to SPIR-V21/// operations.22template <typename Op, typename SPIRVOp>23struct ElementwiseOpPattern : public OpConversionPattern<Op> {24 using OpConversionPattern<Op>::OpConversionPattern;25 26 LogicalResult27 matchAndRewrite(Op op, typename Op::Adaptor adaptor,28 ConversionPatternRewriter &rewriter) const override {29 assert(adaptor.getOperands().size() <= 3);30 Type dstType = this->getTypeConverter()->convertType(op.getType());31 if (!dstType) {32 return rewriter.notifyMatchFailure(33 op->getLoc(),34 llvm::formatv("failed to convert type {0} for SPIR-V", op.getType()));35 }36 37 if (SPIRVOp::template hasTrait<OpTrait::spirv::UnsignedOp>() &&38 !getElementTypeOrSelf(op.getType()).isIndex() &&39 dstType != op.getType()) {40 op.dump();41 return op.emitError("bitwidth emulation is not implemented yet on "42 "unsigned op pattern version");43 }44 rewriter.template replaceOpWithNewOp<SPIRVOp>(op, dstType,45 adaptor.getOperands());46 return success();47 }48};49 50} // namespace spirv51} // namespace mlir52 53#endif // MLIR_CONVERSION_SPIRVCOMMON_PATTERN_H54