Home
last modified time | relevance | path

Searched refs:getDefiningOp (Results 1 – 25 of 167) sorted by relevance

1234567

/llvm-project-15.0.7/mlir/lib/Dialect/Complex/IR/
H A DComplexOps.cpp69 if (auto reOp = getOperand(0).getDefiningOp<ReOp>()) { in fold()
70 if (auto imOp = getOperand(1).getDefiningOp<ImOp>()) { in fold()
88 if (auto createOp = getOperand().getDefiningOp<CreateOp>()) in fold()
102 if (auto createOp = getOperand().getDefiningOp<CreateOp>()) in fold()
115 if (auto sub = getLhs().getDefiningOp<SubOp>()) in fold()
120 if (auto sub = getRhs().getDefiningOp<SubOp>()) in fold()
135 if (auto negOp = getOperand().getDefiningOp<NegOp>()) in fold()
149 if (auto expOp = getOperand().getDefiningOp<ExpOp>()) in fold()
163 if (auto logOp = getOperand().getDefiningOp<LogOp>()) in fold()
/llvm-project-15.0.7/flang/unittests/Optimizer/Builder/Runtime/
H A DNumericTest.cpp20 checkCallOp(exp.getDefiningOp(), fctName, 1, /*addLocArg=*/false); in testGenExponent()
40 checkCallOp(val.getDefiningOp(), fctName, 1, /*addLocArg=*/false); in testGenX()
58 checkCallOp(nearest.getDefiningOp(), fctName, 2, /*addLocArg=*/false); in testGenNearest()
59 auto callOp = mlir::dyn_cast<fir::CallOp>(nearest.getDefiningOp()); in testGenNearest()
61 EXPECT_TRUE(mlir::isa<mlir::arith::SelectOp>(select.getDefiningOp())); in testGenNearest()
62 auto selectOp = mlir::dyn_cast<mlir::arith::SelectOp>(select.getDefiningOp()); in testGenNearest()
64 EXPECT_TRUE(mlir::isa<mlir::arith::CmpFOp>(cmp.getDefiningOp())); in testGenNearest()
65 auto cmpOp = mlir::dyn_cast<mlir::arith::CmpFOp>(cmp.getDefiningOp()); in testGenNearest()
95 checkCallOp(val.getDefiningOp(), fctName, 2, /*addLocArg=*/false); in testGenXI()
H A DCommandTest.cpp16 checkCallOp(result.getDefiningOp(), "_FortranAArgumentCount", /*nbArgs=*/0, in TEST_F()
29 checkCallOp(result.getDefiningOp(), "_FortranAArgumentValue", /*nbArgs=*/3, in TEST_F()
39 checkCallOp(result.getDefiningOp(), "_FortranAArgumentLength", /*nbArgs=*/1, in TEST_F()
52 checkCallOp(result.getDefiningOp(), "_FortranAEnvVariableValue", /*nbArgs=*/4, in TEST_F()
63 checkCallOp(result.getDefiningOp(), "_FortranAEnvVariableLength", in TEST_F()
H A DCharacterTest.cpp43 EXPECT_TRUE(mlir::isa<mlir::arith::CmpIOp>(res.getDefiningOp())); in checkCharCompare1()
83 EXPECT_TRUE(mlir::isa<mlir::arith::CmpIOp>(res.getDefiningOp())); in checkCharCompare2()
84 auto cmpOp = mlir::dyn_cast<mlir::arith::CmpIOp>(res.getDefiningOp()); in checkCharCompare2()
85 checkCallOp(cmpOp.getLhs().getDefiningOp(), fctName, 4, /*addLocArgs=*/false); in checkCharCompare2()
107 checkCallOp(res.getDefiningOp(), fctName, 5, /*addLocArgs=*/false); in checkGenIndex()
170 checkCallOp(res.getDefiningOp(), fctName, 5, /*addLocArgs=*/false); in checkGenScan()
204 checkCallOp(res.getDefiningOp(), fctName, 5, /*addLocArgs=*/false); in checkGenVerify()
H A DReductionTest.cpp18 checkCallOp(all.getDefiningOp(), "_FortranAAll", 2); in TEST_F()
35 checkCallOp(any.getDefiningOp(), "_FortranAAny", 2); in TEST_F()
52 checkCallOp(count.getDefiningOp(), "_FortranACount", 2); in TEST_F()
74 checkCallOp(max.getDefiningOp(), fctName, 3); in testGenMaxVal()
99 checkCallOp(min.getDefiningOp(), fctName, 3); in testGenMinVal()
128 checkCallOp(sum.getDefiningOp(), fctName, 3); in testGenSum()
161 checkCallOp(prod.getDefiningOp(), fctName, 3); in testGenProduct()
193 checkCallOp(prod.getDefiningOp(), fctName, 2); in testGenDotProduct()
/llvm-project-15.0.7/flang/unittests/Optimizer/Builder/
H A DFIRBuilderTest.cpp111 EXPECT_TRUE(mlir::isa<arith::CmpIOp>(res.getDefiningOp())); in TEST_F()
112 auto cmpOp = dyn_cast<arith::CmpIOp>(res.getDefiningOp()); in TEST_F()
122 EXPECT_TRUE(mlir::isa<arith::CmpIOp>(res.getDefiningOp())); in TEST_F()
123 auto cmpOp = dyn_cast<arith::CmpIOp>(res.getDefiningOp()); in TEST_F()
132 EXPECT_TRUE(mlir::isa<fir::ZeroOp>(cst.getDefiningOp())); in TEST_F()
133 auto zeroOp = dyn_cast<fir::ZeroOp>(cst.getDefiningOp()); in TEST_F()
139 EXPECT_TRUE(mlir::isa<fir::ZeroOp>(cst.getDefiningOp())); in TEST_F()
140 zeroOp = dyn_cast<fir::ZeroOp>(cst.getDefiningOp()); in TEST_F()
333 EXPECT_TRUE(mlir::isa<fir::AllocaOp>(var.getDefiningOp())); in TEST_F()
438 zeroFloat.getDefiningOp()); in TEST_F()
[all …]
H A DDoLoopHelperTest.cpp35 EXPECT_TRUE(mlir::isa<mlir::arith::ConstantOp>(value.getDefiningOp())); in checkConstantValue()
36 auto cstOp = dyn_cast<mlir::arith::ConstantOp>(value.getDefiningOp()); in checkConstantValue()
50 EXPECT_TRUE(mlir::isa<arith::SubIOp>(loop.getUpperBound().getDefiningOp())); in TEST_F()
51 auto subOp = dyn_cast<arith::SubIOp>(loop.getUpperBound().getDefiningOp()); in TEST_F()
/llvm-project-15.0.7/flang/lib/Optimizer/CodeGen/
H A DPreCGRewrite.cpp114 if (auto shapeOp = mlir::dyn_cast<fir::ShapeOp>(shapeVal.getDefiningOp())) { in rewriteDynamicShape()
118 mlir::dyn_cast<fir::ShapeShiftOp>(shapeVal.getDefiningOp()); in rewriteDynamicShape()
127 mlir::dyn_cast_or_null<fir::SliceOp>(s.getDefiningOp())) { in rewriteDynamicShape()
167 if (auto shapeOp = mlir::dyn_cast<fir::ShapeOp>(shapeVal.getDefiningOp())) in matchAndRewrite()
170 mlir::dyn_cast<fir::ShapeShiftOp>(shapeVal.getDefiningOp())) in matchAndRewrite()
173 mlir::dyn_cast<fir::ShiftOp>(shapeVal.getDefiningOp())) in matchAndRewrite()
183 mlir::dyn_cast_or_null<fir::SliceOp>(s.getDefiningOp())) { in matchAndRewrite()
229 mlir::dyn_cast<fir::ShapeShiftOp>(shapeVal.getDefiningOp())) in matchAndRewrite()
232 mlir::dyn_cast<fir::ShiftOp>(shapeVal.getDefiningOp())) in matchAndRewrite()
241 mlir::dyn_cast_or_null<fir::SliceOp>(s.getDefiningOp())) { in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/IR/
H A DValue.cpp20 Operation *Value::getDefiningOp() const { in getDefiningOp() function in Value
27 if (auto *op = getDefiningOp()) in getLoc()
34 if (auto *op = getDefiningOp()) in setLoc()
42 if (auto *op = getDefiningOp()) in getParentRegion()
49 if (Operation *op = getDefiningOp()) in getParentBlock()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp48 val.getDefiningOp<InitTensorOp>() || in isEmptyInit()
49 val.getDefiningOp<AllocTensorOp>(); in isEmptyInit()
55 if (auto *def = yieldOp.getOperand(0).getDefiningOp()) { in isSampling()
71 if (auto *def = val.getDefiningOp()) { in isMulChain()
82 if (auto *def = yieldOp.getOperand(0).getDefiningOp()) { in isSumOfMul()
138 op.getInputOperand(other)->get().getDefiningOp()); in matchAndRewrite()
170 auto *acc = prodBlock.getTerminator()->getOperand(0).getDefiningOp(); in matchAndRewrite()
171 auto *sampler = consBlock.getTerminator()->getOperand(0).getDefiningOp(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DBufferOptimizations.cpp37 if (!type || !alloc.getDefiningOp<memref::AllocOp>()) in defaultIsSmallAlloc()
46 return llvm::all_of(alloc.getDefiningOp()->getOperands(), in defaultIsSmallAlloc()
48 return operand.getDefiningOp<memref::RankOp>(); in defaultIsSmallAlloc()
53 unsigned bitwidth = mlir::DataLayout::closest(alloc.getDefiningOp()) in defaultIsSmallAlloc()
139 if (!StateT::shouldHoistOpType(allocValue.getDefiningOp())) in hoist()
141 Operation *definingOp = allocValue.getDefiningOp(); in hoist()
171 Operation *allocOperation = allocValue.getDefiningOp(); in hoist()
346 Operation *allocOp = alloc.getDefiningOp(); in promote()
H A DAllocTensorElimination.cpp76 insertionPointCandidates.push_back(val.getDefiningOp()->getNextNode()); in findValidInsertionPoint()
139 !maybeAllocTensor.front().getDefiningOp<AllocTensorOp>()) in eliminateAllocTensors()
145 findValidInsertionPoint(allocTensor.getDefiningOp(), neededValues); in eliminateAllocTensors()
156 rewriter.replaceOp(allocTensor.getDefiningOp(), replacement); in eliminateAllocTensors()
H A DBufferDeallocation.cpp215 alloc.getDefiningOp<bufferization::AllocationOpInterface>(); in prepare()
219 return alloc.getDefiningOp()->emitError( in prepare()
288 findUnsafeValues(allocValue, allocValue.getDefiningOp()->getBlock()); in introduceClones()
398 Operation *operation = value.getDefiningOp(); in introduceValueCopyForRegionResult()
522 if (alias.getDefiningOp() && in placeDeallocs()
524 *alias.getDefiningOp()))) in placeDeallocs()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp44 return opResult.getDefiningOp(); in getOwnerOfValue()
50 auto bufferizableOp = opResult.getDefiningOp<BufferizableOpInterface>(); in allocationDoesNotEscape()
55 Operation *op = opResult.getDefiningOp(); in allocationDoesNotEscape()
87 shapedValue.getDefiningOp())) { in allocateTensorForShapedValue()
201 if (use->getOwner() != copy->getDefiningOp()) in resolveTensorOpOperandConflicts()
313 b.setInsertionPointAfter(value.getDefiningOp()); in setInsertionPointAfter()
321 if (Operation *op = result.getDefiningOp()) in getAliasingOpOperand()
430 Operation *op = value.getDefiningOp(); in findLastPrecedingWrite()
497 if (!tensor.getDefiningOp<AllocTensorOp>()) in isTensorYielded()
556 if (auto toTensorOp = value.getDefiningOp<bufferization::ToTensorOp>()) in getBuffer()
[all …]
H A DBufferizationOps.cpp90 auto memrefToTensor = toMemref.getTensor().getDefiningOp<ToTensorOp>(); in foldToMemrefToTensorPair()
335 auto allocTensorOp = dimOp.getSource().getDefiningOp<AllocTensorOp>(); in matchAndRewrite()
533 if (auto toMemref = getMemref().getDefiningOp<ToMemrefOp>()) in fold()
548 auto memrefToTensorOp = dimOp.getSource().getDefiningOp<ToTensorOp>(); in matchAndRewrite()
569 if (auto memrefToTensor = getTensor().getDefiningOp<ToTensorOp>()) in fold()
584 toMemref.getOperand().getDefiningOp<tensor::CastOp>(); in matchAndRewrite()
619 auto toMemref = load.getMemref().getDefiningOp<ToMemrefOp>(); in matchAndRewrite()
635 auto castOp = dimOp.getSource().getDefiningOp<ToMemrefOp>(); in matchAndRewrite()
/llvm-project-15.0.7/flang/lib/Optimizer/Transforms/
H A DAffinePromotion.cpp73 op->dump(); coordinate.getDefiningOp()->dump()); in analyzeCoordinate()
109 if (auto acoOp = memref.getDefiningOp<ArrayCoorOp>()) { in analyzeReference()
122 if (auto coOp = memref.getDefiningOp<CoordinateOp>()) { in analyzeReference()
206 if (auto op = value.getDefiningOp<mlir::arith::SubIOp>()) in toAffineExpr()
211 if (auto op = value.getDefiningOp<mlir::arith::AddIOp>()) in toAffineExpr()
214 if (auto op = value.getDefiningOp<mlir::arith::MulIOp>()) in toAffineExpr()
217 if (auto op = value.getDefiningOp<mlir::arith::RemUIOp>()) in toAffineExpr()
220 if (auto op = value.getDefiningOp<mlir::arith::ConstantOp>()) in toAffineExpr()
387 if (auto shape = acoOp.getShape().getDefiningOp<ShapeOp>()) in populateIndexArgs()
391 if (auto slice = acoOp.getShape().getDefiningOp<SliceOp>()) in populateIndexArgs()
[all …]
H A DAnnotateConstant.cpp32 opnd.getDefiningOp())) { in runOnOperation()
36 opnd.getDefiningOp())) { in runOnOperation()
H A DArrayValueCopy.cpp279 if (auto *op = val.getDefiningOp()) { in collectArrayMentionFrom()
480 auto *op = v.getDefiningOp(); in mutuallyExclusiveSliceRange()
482 op = conv.getValue().getDefiningOp(); in mutuallyExclusiveSliceRange()
499 if ((addi.getLhs().getDefiningOp() == op1 && in mutuallyExclusiveSliceRange()
501 (addi.getRhs().getDefiningOp() == op1 && in mutuallyExclusiveSliceRange()
505 if (subi.getLhs().getDefiningOp() == op2 && in mutuallyExclusiveSliceRange()
666 return amend.getMemref().getDefiningOp(); in amendingAccess()
712 conflicts.insert(st.getOriginal().getDefiningOp()); in construct()
716 auto *ld = st.getOriginal().getDefiningOp(); in construct()
791 auto *shapeOp = shape.getDefiningOp(); in getAdjustedExtents()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPU.cpp172 if (auto def = value.getDefiningOp<arith::ConstantIndexOp>()) in isConstantOne()
318 if (auto op = upperBound.getDefiningOp<arith::ConstantIndexOp>()) { in deriveStaticUpperBound()
322 if (auto minOp = upperBound.getDefiningOp<AffineMinOp>()) { in deriveStaticUpperBound()
331 if (auto multiplyOp = upperBound.getDefiningOp<arith::MulIOp>()) { in deriveStaticUpperBound()
334 .getDefiningOp())) in deriveStaticUpperBound()
337 .getDefiningOp())) { in deriveStaticUpperBound()
420 if (auto constOp = val.getDefiningOp<arith::ConstantOp>()) in processParallelLoop()
466 !isa_and_nonnull<arith::ConstantOp>(lowerBound.getDefiningOp())) in processParallelLoop()
470 !isa_and_nonnull<arith::ConstantOp>(step.getDefiningOp())) in processParallelLoop()
476 isa_and_nonnull<arith::ConstantOp>(upperBound.getDefiningOp()); in processParallelLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DLoopPipelining.cpp88 forOp.getUpperBound().getDefiningOp<arith::ConstantIndexOp>(); in initializeLoopInfo()
90 forOp.getLowerBound().getDefiningOp<arith::ConstantIndexOp>(); in initializeLoopInfo()
91 auto stepCst = forOp.getStep().getDefiningOp<arith::ConstantIndexOp>(); in initializeLoopInfo()
132 Operation *def = operand.getDefiningOp(); in initializeLoopInfo()
185 Operation *def = operand.get().getDefiningOp(); in analyzeCrossStageValues()
214 Operation *def = retVal.value().getDefiningOp(); in createKernelLoop()
299 Operation *dep = ret.getDefiningOp(); in createKernel()
313 Operation *def = operand.get().getDefiningOp(); in createKernel()
365 Operation *def = retVal.value().getDefiningOp(); in createKernel()
H A DParallelLoopTiling.cpp100 dyn_cast_or_null<arith::ConstantIndexOp>(lowerBound.getDefiningOp()); in tileParallelLoop()
102 dyn_cast_or_null<arith::ConstantIndexOp>(upperBound.getDefiningOp()); in tileParallelLoop()
104 dyn_cast_or_null<arith::ConstantIndexOp>(step.getDefiningOp()); in tileParallelLoop()
106 cast<arith::ConstantIndexOp>(tileSizeConstant.getDefiningOp()).value(); in tileParallelLoop()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusion.cpp81 opOperand->get().getDefiningOp())) in getShapeDefiningLoopRange()
182 Operation *shapeProducingOp = shapedOperand.getDefiningOp(); in getRangeFromOperandShape()
357 auto subView = consumerOpOperand.get().getDefiningOp<memref::SubViewOp>(); in fuseProducerOfBuffer()
385 if (auto linalgOp = tensor.getDefiningOp<LinalgOp>()) { in getProducerOfTensor()
389 if (auto sliceOp = tensor.getDefiningOp<tensor::ExtractSliceOp>()) { in getProducerOfTensor()
394 if (auto forOp = blockArg.getDefiningOp<scf::ForOp>()) { in getProducerOfTensor()
429 auto sliceOp = inputTensor.getDefiningOp<tensor::ExtractSliceOp>(); in fuseProducerOfTensor()
H A DPromotion.cpp56 if (auto cst = allocSize.getDefiningOp<arith::ConstantIndexOp>()) { in allocBuffer()
110 auto viewOp = cast<memref::ViewOp>(fullLocalView.getDefiningOp()); in defaultDeallocBufferCallBack()
157 Operation *op = opOperand->get().getDefiningOp(); in LinalgOpInstancePromotionOptions()
262 cast<memref::SubViewOp>(v.second.getDefiningOp()); in promoteSubViews()
301 b, cast<memref::SubViewOp>(v.second.getDefiningOp()), in promoteSubViews()
370 isa_and_nonnull<memref::SubViewOp>(opOperand->get().getDefiningOp()); in promoteSubviewsPrecondition()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp57 if (auto c = mask.getDefiningOp<arith::ConstantOp>()) { in get1DMaskFormat()
75 } else if (auto m = mask.getDefiningOp<ConstantMaskOp>()) { in get1DMaskFormat()
917 maybeContraction.getDefiningOp()); in matchAndRewrite()
921 contractionOp.getAcc().getDefiningOp())) { in matchAndRewrite()
1078 if (!extractOp.getVector().getDefiningOp<ExtractOp>()) in foldExtractOpFromExtractChain()
1148 nextInsertOp = v.getDefiningOp<vector::InsertOp>(); in updateStateForNextIteration()
1889 auto v1Splat = op.getV1().getDefiningOp<SplatOp>(); in matchAndRewrite()
1890 auto v2Splat = op.getV2().getDefiningOp<SplatOp>(); in matchAndRewrite()
2034 auto dstSplat = op.getDest().getDefiningOp<SplatOp>(); in matchAndRewrite()
2746 auto splat = op.getVector().getDefiningOp<SplatOp>(); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/test/lib/Dialect/Linalg/
H A DTestLinalgElementwiseFusion.cpp130 producer.getDefiningOp<tensor::CollapseShapeOp>()) { in runOnOperation()
131 if (!collapseOp.getSrc().getDefiningOp<linalg::LinalgOp>()) { in runOnOperation()
167 if (isa<tensor::ExpandShapeOp>(producer.getDefiningOp())) { in runOnOperation()

1234567