Home
last modified time | relevance | path

Searched refs:SelectOp (Results 1 – 25 of 55) sorted by relevance

123

/llvm-project-15.0.7/mlir/lib/Conversion/ComplexToStandard/
H A DComplexToStandard.cpp586 lhsReal = b.create<arith::SelectOp>( in matchAndRewrite()
591 lhsImag = b.create<arith::SelectOp>( in matchAndRewrite()
596 rhsReal = b.create<arith::SelectOp>( in matchAndRewrite()
601 rhsImag = b.create<arith::SelectOp>( in matchAndRewrite()
617 rhsReal = b.create<arith::SelectOp>( in matchAndRewrite()
622 rhsImag = b.create<arith::SelectOp>( in matchAndRewrite()
627 lhsReal = b.create<arith::SelectOp>( in matchAndRewrite()
632 lhsImag = b.create<arith::SelectOp>( in matchAndRewrite()
660 lhsReal = b.create<arith::SelectOp>( in matchAndRewrite()
686 real = b.create<arith::SelectOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/
H A DPolynomialApproximation.cpp187 return builder.create<arith::SelectOp>( in min()
194 return builder.create<arith::SelectOp>( in max()
546 Value res = builder.create<arith::SelectOp>( in matchAndRewrite()
676 Value aproximation = builder.create<arith::SelectOp>( in logMatchAndRewrite()
678 builder.create<arith::SelectOp>( in logMatchAndRewrite()
754 Value approximation = builder.create<arith::SelectOp>( in matchAndRewrite()
994 expY = builder.create<arith::SelectOp>( in matchAndRewrite()
996 builder.create<arith::SelectOp>( in matchAndRewrite()
998 builder.create<arith::SelectOp>( in matchAndRewrite()
1000 builder.create<arith::SelectOp>( in matchAndRewrite()
[all …]
H A DExpandPatterns.cpp52 rewriter.replaceOpWithNewOp<arith::SelectOp>(op, cmpRes, positiveRes, in convertTanhOp()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToArith/
H A DTosaToArith.cpp104 rewriter.create<arith::SelectOp>(loc, positive, roundUp, roundDown); in matchAndRewrite()
109 rewriter.create<arith::SelectOp>(loc, valid, val, multiply64); in matchAndRewrite()
174 rewriter.create<arith::SelectOp>(loc, shiftOver32, zero32, shiftHighL); in matchAndRewrite()
176 rewriter.create<arith::SelectOp>(loc, shiftOver32, shiftHighR, zero32); in matchAndRewrite()
185 rewriter.create<arith::SelectOp>(loc, valuePositive, one32, negOne32); in matchAndRewrite()
187 rewriter.create<arith::SelectOp>(loc, shiftOver32, roundDir, zero32); in matchAndRewrite()
204 roundBit = rewriter.create<arith::SelectOp>(loc, roundHighBits, zero32, in matchAndRewrite()
221 roundBit = rewriter.create<arith::SelectOp>(loc, roundHighBits, roundBit, in matchAndRewrite()
230 low32 = rewriter.create<arith::SelectOp>(loc, shiftOver32, zero32, low32); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/Transforms/
H A DExpandOps.cpp42 rewriter.replaceOpWithNewOp<arith::SelectOp>(op, compare, zero, plusOne); in matchAndRewrite()
64 Value x = rewriter.create<arith::SelectOp>(loc, compare, minusOne, plusOne); in matchAndRewrite()
93 rewriter.replaceOpWithNewOp<arith::SelectOp>(op, compareRes, posRes, in matchAndRewrite()
116 Value x = rewriter.create<arith::SelectOp>(loc, compare, plusOne, minusOne); in matchAndRewrite()
143 rewriter.replaceOpWithNewOp<arith::SelectOp>(op, compareRes, negRes, in matchAndRewrite()
165 Value select = rewriter.create<arith::SelectOp>(loc, cmp, lhs, rhs); in matchAndRewrite()
170 rewriter.replaceOpWithNewOp<arith::SelectOp>(op, isNaN, rhs, select); in matchAndRewrite()
186 rewriter.replaceOpWithNewOp<arith::SelectOp>(op, cmp, lhs, rhs); in matchAndRewrite()
H A DBufferizableOpInterfaceImpl.cpp118 arith::SelectOp> {
143 auto selectOp = cast<arith::SelectOp>(op); in bufferize()
181 replaceOpWithNewBufferizedOp<arith::SelectOp>( in bufferize()
199 SelectOp::attachInterface<SelectOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/
H A DCoversionUtils.h32 arith::SelectOp clampHelper(Location loc, Value arg, arith::ConstantOp min, in clampHelper()
37 rewriter.create<arith::SelectOp>(loc, smallerThanMin, min, arg); in clampHelper()
39 return rewriter.create<arith::SelectOp>(loc, largerThanMax, max, minOrArg); in clampHelper()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToOpenMP/
H A DSCFToOpenMP.cpp75 llvm::is_one_of<SelectOpTy, arith::SelectOp, LLVM::SelectOp>::value, in matchSelectReduction()
293 if (matchSelectReduction<arith::CmpFOp, arith::SelectOp>( in declareReduction()
296 matchSelectReduction<LLVM::FCmpOp, LLVM::SelectOp>( in declareReduction()
302 if (matchSelectReduction<arith::CmpIOp, arith::SelectOp>( in declareReduction()
305 matchSelectReduction<LLVM::ICmpOp, LLVM::SelectOp>( in declareReduction()
314 if (matchSelectReduction<arith::CmpIOp, arith::SelectOp>( in declareReduction()
317 matchSelectReduction<LLVM::ICmpOp, LLVM::SelectOp>( in declareReduction()
/llvm-project-15.0.7/mlir/lib/Conversion/ArithmeticToSPIRV/
H A DArithmeticToSPIRV.cpp196 class SelectOpPattern final : public OpConversionPattern<arith::SelectOp> {
198 using OpConversionPattern<arith::SelectOp>::OpConversionPattern;
200 matchAndRewrite(arith::SelectOp op, OpAdaptor adaptor,
475 return builder.create<spirv::SelectOp>(loc, type, isPositive, abs, absNegate); in emulateSignedRemainder()
585 rewriter.template replaceOpWithNewOp<spirv::SelectOp>( in matchAndRewrite()
606 rewriter.template replaceOpWithNewOp<spirv::SelectOp>( in matchAndRewrite()
633 rewriter.replaceOpWithNewOp<spirv::SelectOp>(op, dstType, isOne, one, zero); in matchAndRewrite()
841 SelectOpPattern::matchAndRewrite(arith::SelectOp op, OpAdaptor adaptor, in matchAndRewrite()
843 rewriter.replaceOpWithNewOp<spirv::SelectOp>(op, adaptor.getCondition(), in matchAndRewrite()
/llvm-project-15.0.7/flang/unittests/Optimizer/Builder/Runtime/
H A DNumericTest.cpp61 EXPECT_TRUE(mlir::isa<mlir::arith::SelectOp>(select.getDefiningOp())); in testGenNearest()
62 auto selectOp = mlir::dyn_cast<mlir::arith::SelectOp>(select.getDefiningOp()); in testGenNearest()
/llvm-project-15.0.7/mlir/lib/Conversion/MathToSPIRV/
H A DMathToSPIRV.cpp162 rewriter.replaceOpWithNewOp<spirv::SelectOp>(countOp, cmp, subInput, in matchAndRewrite()
232 rewriter.replaceOpWithNewOp<spirv::SelectOp>(powfOp, lessThan, negate, pow); in matchAndRewrite()
267 auto select = rewriter.create<spirv::SelectOp>(loc, greater, one, zero); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/IR/
H A DArithmeticOps.cpp1749 struct SelectI1Simplify : public OpRewritePattern<arith::SelectOp> {
1750 using OpRewritePattern<arith::SelectOp>::OpRewritePattern;
1752 LogicalResult matchAndRewrite(arith::SelectOp op, in matchAndRewrite()
1772 struct SelectToExtUI : public OpRewritePattern<arith::SelectOp> {
1773 using OpRewritePattern<arith::SelectOp>::OpRewritePattern;
1775 LogicalResult matchAndRewrite(arith::SelectOp op, in matchAndRewrite()
1805 void arith::SelectOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns()
1810 OpFoldResult arith::SelectOp::fold(ArrayRef<Attribute> operands) { in fold()
1851 ParseResult SelectOp::parse(OpAsmParser &parser, OperationState &result) { in parse()
1874 void arith::SelectOp::print(OpAsmPrinter &p) { in print()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp61 return rewriter.create<arith::SelectOp>(loc, cmp, args[0], neg); in createLinalgBodyCalculationForElementwiseOp()
330 if (isa<tosa::SelectOp>(op)) { in createLinalgBodyCalculationForElementwiseOp()
333 return rewriter.create<arith::SelectOp>(loc, args[0], args[1], args[2]); in createLinalgBodyCalculationForElementwiseOp()
340 return rewriter.create<arith::SelectOp>(loc, predicate, args[0], args[1]); in createLinalgBodyCalculationForElementwiseOp()
346 return rewriter.create<arith::SelectOp>(loc, predicate, args[0], args[1]); in createLinalgBodyCalculationForElementwiseOp()
522 rewriter.create<arith::SelectOp>(loc, negative, subbed, added); in createLinalgBodyCalculationForElementwiseOp()
1491 rewriter.create<arith::SelectOp>(loc, yPred, oneVal, zeroVal); in matchAndRewrite()
1493 rewriter.create<arith::SelectOp>(loc, xPred, oneVal, zeroVal); in matchAndRewrite()
2024 auto resultMax = rewriter.create<arith::SelectOp>( in matchAndRewrite()
2026 auto resultIndex = rewriter.create<arith::SelectOp>( in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/ArithmeticToLLVM/
H A DArithmeticToLLVM.cpp70 VectorConvertToLLVMPattern<arith::SelectOp, LLVM::SelectOp>;
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaCanonicalizations.cpp120 LogicalResult SelectOp::canonicalize(SelectOp op, PatternRewriter &rewriter) { in canonicalize()
502 OpFoldResult tosa::SelectOp::fold(ArrayRef<Attribute> operands) { in fold()
/llvm-project-15.0.7/mlir/lib/Conversion/ShapeToStandard/
H A DShapeToStandard.cpp110 Value dim = b.create<arith::SelectOp>( in getBroadcastedDim()
147 maxRank = lb.create<arith::SelectOp>(rankIsGreater, v, maxRank); in matchAndRewrite()
262 maxRank = lb.create<arith::SelectOp>(rankIsGreater, v, maxRank); in matchAndRewrite()
622 Value index = b.create<arith::SelectOp>(indexIsNegative, add, originalIndex); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/LLVMIR/Transforms/
H A DOptimizeForNVVM.cpp79 rewriter.create<LLVM::SelectOp>(loc, f32Type, pred, approx, refined); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DExpandOps.cpp69 Value select = bodyBuilder.create<arith::SelectOp>(loc, cmp, lhs, rhs); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/Utils/
H A DUtils.cpp116 return b.create<arith::SelectOp>(loc, cmp, lhs, rhs); in select()
/llvm-project-15.0.7/mlir/lib/Conversion/GPUToNVVM/
H A DWmmaOpsToNvvm.cpp289 Value sel = builder.create<LLVM::SelectOp>(loc, cmp, lhs, rhs); in createMinMaxF()
296 return builder.create<LLVM::SelectOp>(loc, isNan, nan, sel); in createMinMaxF()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp97 Value result = builder.create<arith::SelectOp>( in visitModExpr()
135 builder.create<arith::SelectOp>(loc, negative, negatedDecremented, lhs); in visitFloorDivExpr()
139 Value result = builder.create<arith::SelectOp>(loc, negative, in visitFloorDivExpr()
176 builder.create<arith::SelectOp>(loc, nonPositive, negated, decremented); in visitCeilDivExpr()
182 Value result = builder.create<arith::SelectOp>( in visitCeilDivExpr()
/llvm-project-15.0.7/flang/lib/Lower/
H A DIntrinsicCall.cpp2593 return builder.create<mlir::arith::SelectOp>(loc, cmp, diff, zero); in genDim()
2600 return builder.create<mlir::arith::SelectOp>(loc, cmp, diff, zero); in genDim()
2938 return builder.create<mlir::arith::SelectOp>(loc, lenIsZero, zero, res2); in genIbits()
3146 return builder.create<mlir::arith::SelectOp>(loc, shiftIsLarge, zero, sel); in genIshft()
3196 auto unchanged = builder.create<mlir::arith::SelectOp>(loc, hasUnchanged, in genIshftc()
3212 return builder.create<mlir::arith::SelectOp>(loc, shiftIsNop, I, res); in genIshftc()
3397 return builder.create<mlir::arith::SelectOp>(loc, mustAddP, remPlusP, in genModulo()
3414 return builder.create<mlir::arith::SelectOp>(loc, mustAddP, remPlusP, in genModulo()
3861 return builder.create<mlir::arith::SelectOp>(loc, cmp, neg, abs); in genSign()
3949 return builder.create<mlir::arith::SelectOp>(loc, dimIsEmpty, one, lb); in computeLBOUND()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/
H A DAsyncParallelFor.cpp407 auto lb = b.create<arith::SelectOp>(isBlockFirstCoord[loopIdx], in createParallelComputeFunction()
410 auto ub = b.create<arith::SelectOp>(isBlockLastCoord[loopIdx], in createParallelComputeFunction()
837 scalingFactor = b.create<arith::SelectOp>(inBracket, bracketScalingFactor, in matchAndRewrite()
/llvm-project-15.0.7/flang/lib/Optimizer/Builder/
H A DCharacter.cpp414 return builder.create<mlir::arith::SelectOp>(loc, cmp, a, b); in genMin()
515 builder.create<mlir::arith::SelectOp>(loc, cdt, zero, substringLen); in createSubstring()
553 auto result = builder.create<mlir::arith::SelectOp>( in createLenTrim()
/llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/IR/
H A DSPIRVCanonicalization.cpp332 auto selectOp = rewriter.create<spirv::SelectOp>( in matchAndRewrite()

123