mlir/lib/Conversion/SPIRVCommon/Pattern.h

   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-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8
   9 #ifndef MLIR_CONVERSION_SPIRVCOMMON_PATTERN_H
  10 #define MLIR_CONVERSION_SPIRVCOMMON_PATTERN_H
  11
  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
  17 namespace mlir {
  18 namespace spirv {
  19
  20 /// Converts elementwise unary, binary and ternary standard operations to SPIR-V
  21 /// operations.
  22 template <typename Op, typename SPIRVOp>
  23 struct ElementwiseOpPattern : public OpConversionPattern<Op> {
  24   using OpConversionPattern<Op>::OpConversionPattern;
  25
  26   LogicalResult
  27   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 spirv
  51 } // namespace mlir
  52
  53 #endif // MLIR_CONVERSION_SPIRVCOMMON_PATTERN_H