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 matchAndRewriteElementwiseOpPattern27 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 54