Home
last modified time | relevance | path

Searched refs:MulIOp (Results 1 – 25 of 35) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DLoopRangeFolding.cpp48 if (!isa<arith::AddIOp, arith::MulIOp>(user)) in runOnOperation()
69 } else if (isa<arith::MulIOp>(user)) { in runOnOperation()
H A DParallelLoopTiling.cpp73 newSteps.push_back(b.create<arith::MulIOp>(op.getLoc(), std::get<0>(step), in tileParallelLoop()
150 op.getLoc(), b.create<arith::MulIOp>(op.getLoc(), innerIV, innerStep), in tileParallelLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp75 Value tmp1 = create<arith::MulIOp>(int32Type, tidZ, dimY); in rewrite()
77 Value tmp3 = create<arith::MulIOp>(int32Type, tmp2, dimX); in rewrite()
78 Value tmp4 = create<arith::MulIOp>(int32Type, dimX, dimY); in rewrite()
80 Value workgroupSize = create<arith::MulIOp>(int32Type, tmp4, dimZ); in rewrite()
224 : getFactory<arith::MulIOp>(); in getFactory()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/
H A DUtils.cpp491 boundsBuilder.create<arith::MulIOp>(loc, tripCountEvenMultiple, step)); in loopUnrollByFactor()
494 boundsBuilder.create<arith::MulIOp>(loc, step, unrollFactorCst); in loopUnrollByFactor()
528 auto stride = b.create<arith::MulIOp>( in loopUnrollByFactor()
578 : insideLoopBuilder.create<arith::MulIOp>(loc, inductionVar, step); in normalizeLoop()
631 builder.create<arith::MulIOp>(loc, upperBound, loop.getUpperBound()); in coalesceLoops()
700 newUpperBound = outsideBuilder.create<arith::MulIOp>( in collapseParallelLoops()
841 forOp.setStep(b.create<arith::MulIOp>(forOp.getLoc(), originalStep, factor)); in stripmineSink()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToArith/
H A DTosaToArith.cpp88 rewriter.create<arith::MulIOp>(loc, value64, multiplier64); in matchAndRewrite()
154 rewriter.create<arith::MulIOp>(loc, value64, multiplier64); in matchAndRewrite()
160 Value low32 = rewriter.create<arith::MulIOp>(loc, value32, multiplier32); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp56 if (isa<arith::MulFOp>(def) || isa<arith::MulIOp>(def)) { in isSampling()
72 if (isa<arith::MulFOp>(def) || isa<arith::MulIOp>(def)) in isMulChain()
H A DSparsification.cpp670 return builder.create<arith::MulIOp>( in genAffine()
887 Value mul = builder.create<arith::MulIOp>(loc, size, p); in genAddress()
1217 step = builder.create<arith::MulIOp>(loc, vscale, step); in genFor()
/llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/
H A DAsyncParallelFor.cpp313 tripCount = b.create<arith::MulIOp>(tripCount, tripCounts[i]); in createParallelComputeFunction()
317 Value blockFirstIndex = b.create<arith::MulIOp>(blockIndex, blockSize); in createParallelComputeFunction()
380 lowerBounds[loopIdx], b.create<arith::MulIOp>(iv, steps[loopIdx])); in createParallelComputeFunction()
748 tripCount = b.create<arith::MulIOp>(tripCount, tripCounts[i]); in matchAndRewrite()
892 Value alignedBlockSize = b.create<arith::MulIOp>( in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DPromotion.cpp69 Value mul = b.createOrFold<arith::MulIOp>( in allocBuffer()
92 allocSize = b.createOrFold<arith::MulIOp>(allocSize, size.value()); in defaultAllocBufferCallBack()
H A DSplitReduction.cpp56 if (isa<arith::MulIOp>(op)) in getNeutralElement()
H A DVectorization.cpp130 .Case<arith::MulIOp, arith::MulFOp>( in getCombinerOpKind()
1354 if (!mulOp || !isa<arith::MulIOp, arith::MulFOp>(mulOp)) in Conv1DNwcGenerator()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DExpandOps.cpp113 stride = rewriter.create<arith::MulIOp>(loc, stride, size); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/Utils/
H A DUtils.cpp102 return b.create<arith::MulIOp>(loc, lhs, rhs); in mul()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp132 return rewriter.create<arith::MulIOp>(loc, resultTypes, a, b); in createLinalgBodyCalculationForElementwiseOp()
750 return rewriter.create<arith::MulIOp>(loc, args); in createLinalgBodyCalculationForReduceOp()
1450 Value y = rewriter.create<arith::MulIOp>(loc, inY, yStride); in matchAndRewrite()
1451 Value x = rewriter.create<arith::MulIOp>(loc, inX, xStride); in matchAndRewrite()
1594 y0x0 = rewriter.create<arith::MulIOp>(loc, y0x0, leftPart); in matchAndRewrite()
1595 y0x1 = rewriter.create<arith::MulIOp>(loc, y0x1, rightPart); in matchAndRewrite()
1598 y1x0 = rewriter.create<arith::MulIOp>(loc, y1x0, leftPart); in matchAndRewrite()
1599 y1x1 = rewriter.create<arith::MulIOp>(loc, y1x1, rightPart); in matchAndRewrite()
1604 topAcc = rewriter.create<arith::MulIOp>(loc, topAcc, topPart); in matchAndRewrite()
1605 bottomAcc = rewriter.create<arith::MulIOp>(loc, bottomAcc, bottomPart); in matchAndRewrite()
[all …]
H A DTosaToLinalgNamed.cpp90 Value dilated = builder.create<arith::MulIOp>(dilation, subOne); in getConvOutputDim()
876 Value count = rewriter.create<arith::MulIOp>(loc, kH3, kW3); in matchAndRewrite()
897 rewriter.create<arith::MulIOp>(loc, accETy, countI, inputZp); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPU.cpp269 id = builder.create<arith::MulIOp>(rootForOp.getLoc(), step, id); in createLaunch()
331 if (auto multiplyOp = upperBound.getDefiningOp<arith::MulIOp>()) { in deriveStaticUpperBound()
/llvm-project-15.0.7/mlir/lib/Conversion/ArithmeticToLLVM/
H A DArithmeticToLLVM.cpp27 using MulIOpLowering = VectorConvertToLLVMPattern<arith::MulIOp, LLVM::MulOp>;
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDistribute.cpp58 Value storeOffset = rewriter.create<arith::MulIOp>( in rewriteWarpOpToScfFor()
114 Value loadOffset = rewriter.create<arith::MulIOp>( in rewriteWarpOpToScfFor()
/llvm-project-15.0.7/flang/lib/Optimizer/Builder/
H A DCharacter.cpp229 len = builder.create<mlir::arith::MulIOp>( in toScalarCharacter()
333 builder.create<mlir::arith::MulIOp>(loc, kindBytes, castCount); in createCopy()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DAffineAnalysis.cpp60 .Case([](arith::MulIOp) { return arith::AtomicRMWKind::muli; }) in getSupportedReduction() argument
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Utils/
H A DMerger.cpp976 if (isa<arith::MulIOp>(def)) in buildTensorExp()
1157 return rewriter.create<arith::MulIOp>(loc, v0, v1); in buildExp()
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Demitter.py438 return arith.MulIOp(lhs, rhs).result
/llvm-project-15.0.7/flang/lib/Optimizer/Transforms/
H A DAffinePromotion.cpp214 if (auto op = value.getDefiningOp<mlir::arith::MulIOp>()) in toAffineExpr()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp131 !isAddMul<arith::AddIOp, arith::MulIOp>(linalgOp->getRegion(0).front()) && in isContractionInterfaceImpl()
/llvm-project-15.0.7/flang/lib/Lower/
H A DConvertExpr.cpp1207 GENBIN(Multiply, Integer, mlir::arith::MulIOp) in GENBIN()
1371 builder.create<mlir::arith::MulIOp>(loc, bufferSize, ratio); in genval()
2012 builder.create<mlir::arith::MulIOp>(loc, delta, diff); in genOffsetAndCoordinateOp()
2015 delta = builder.create<mlir::arith::MulIOp>(loc, delta, ext); in genOffsetAndCoordinateOp()
5367 GENBIN(Multiply, Integer, mlir::arith::MulIOp) in GENBIN()
5709 auto prod = builder.create<mlir::arith::MulIOp>( in genSliceIndices()
6351 auto newSz = builder.create<mlir::arith::MulIOp>(loc, needed, two); in growBuffer()
6398 eleSz = builder.create<mlir::arith::MulIOp>(loc, bytes, length); in copyNextArrayCtorSection()
6417 off = builder.create<mlir::arith::MulIOp>(loc, off, eleSz); in copyNextArrayCtorSection()
6432 arrSz = builder.create<mlir::arith::MulIOp>(loc, arrSz, ext); in copyNextArrayCtorSection()
[all …]

12