| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | LoopCanonicalization.cpp | 89 auto blockArg = dimOp.getSource().template dyn_cast<BlockArgument>(); in matchAndRewrite() local
90 if (!blockArg) in matchAndRewrite()
92 auto forOp = dyn_cast<ForOp>(blockArg.getParentBlock()->getParentOp()); in matchAndRewrite()
95 if (!isShapePreserving(forOp, blockArg.getArgNumber() - 1)) in matchAndRewrite()
98 Value initArg = forOp.getOpOperandForRegionIterArg(blockArg).get(); in matchAndRewrite()
|
| H A D | TileUsingInterface.cpp | 347 while (auto blockArg = v.dyn_cast<BlockArgument>()) { in getFusableProducer() local
348 auto loopOp = dyn_cast<scf::ForOp>(blockArg.getOwner()->getParentOp()); in getFusableProducer()
351 v = loopOp.getOpOperandForRegionIterArg(blockArg).get(); in getFusableProducer()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Detensorize.cpp | 410 for (auto &blockArg : blockArgsToDetensor) { in compute() local
411 Block *block = blockArg.getParentBlock(); in compute()
423 blockOperands.isOperandProduced(blockArg.getArgNumber())) in compute()
427 blockOperands[blockArg.getArgNumber()].getDefiningOp(); in compute()
433 blockArgsToRemove.insert(blockArg); in compute()
439 for (auto &blockArg : blockArgsToRemove) { in compute() local
440 blockArgsToDetensor.erase(blockArg); in compute()
|
| H A D | Fusion.cpp | 393 if (auto blockArg = tensor.dyn_cast<BlockArgument>()) { in getProducerOfTensor() local
394 if (auto forOp = blockArg.getDefiningOp<scf::ForOp>()) { in getProducerOfTensor()
395 tensor = *(forOp.getIterOperands().begin() + blockArg.getArgNumber()); in getProducerOfTensor()
|
| /llvm-project-15.0.7/mlir/test/lib/IR/ |
| H A D | TestPrintDefUse.cpp | 37 auto blockArg = operand.cast<BlockArgument>(); in runOnOperation() local
39 << blockArg.getArgNumber() << "\n"; in runOnOperation()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/ |
| H A D | BufferDeallocation.cpp | 318 LogicalResult introduceBlockArgCopy(BlockArgument blockArg) { in introduceBlockArgCopy() argument
322 Block *block = blockArg.getOwner(); in introduceBlockArgCopy()
332 Value sourceValue = operands[blockArg.getArgNumber()]; in introduceBlockArgCopy()
341 operands.slice(blockArg.getArgNumber(), 1).assign(*clone); in introduceBlockArgCopy()
356 regionInterface, argRegion->getParentOp()->getRegions(), blockArg, in introduceBlockArgCopy()
380 llvm::find(it->getSuccessorInputs(), blockArg).getIndex() + in introduceBlockArgCopy()
|
| /llvm-project-15.0.7/mlir/lib/Transforms/ |
| H A D | ViewOpGraph.cpp | 245 for (BlockArgument &blockArg : block.getArguments()) in processBlock()
246 valueToNode[blockArg] = emitNodeStmt(getLabel(blockArg)); in processBlock()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaInferShapes.cpp | 49 auto blockArg = frontBlock.getArgument(i - 1); in propagateShapesToTosaIf() local
50 auto oldType = blockArg.getType().cast<ShapedType>(); in propagateShapesToTosaIf()
54 blockArg.setType(newType); in propagateShapesToTosaIf()
|
| /llvm-project-15.0.7/mlir/lib/AsmParser/ |
| H A D | AsmParserState.cpp | 262 void AsmParserState::addDefinition(BlockArgument blockArg, SMLoc location) { in addDefinition() argument
263 auto it = impl->blocksToIdx.find(blockArg.getOwner()); in addDefinition()
267 unsigned argIdx = blockArg.getArgNumber(); in addDefinition()
|
| /llvm-project-15.0.7/mlir/lib/Analysis/ |
| H A D | Liveness.cpp | 283 auto blockArg = value.cast<BlockArgument>(); in print() local
284 os << "arg" << blockArg.getArgNumber() << "@" in print()
285 << blockIds[blockArg.getOwner()]; in print()
|
| H A D | SliceAnalysis.cpp | 95 } else if (auto blockArg = operand.dyn_cast<BlockArgument>()) { in getBackwardSliceImpl() local
96 Block *block = blockArg.getOwner(); in getBackwardSliceImpl()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | Dominance.cpp | 300 if (auto blockArg = a.dyn_cast<BlockArgument>()) in properlyDominates() local
301 return dominates(blockArg.getOwner(), b->getBlock()); in properlyDominates()
|
| /llvm-project-15.0.7/mlir/lib/Target/SPIRV/Deserialization/ |
| H A D | Deserializer.cpp | 1583 BlockArgument blockArg = curBlock->addArgument(blockArgType, unknownLoc); in processPhi() local
1584 valueMap[operands[1]] = blockArg; in processPhi()
1586 << "[phi] created block argument " << blockArg in processPhi()
1750 for (BlockArgument blockArg : block->getArguments()) { in structurize() local
1752 newBlock->addArgument(blockArg.getType(), blockArg.getLoc()); in structurize()
1753 mapper.map(blockArg, newArg); in structurize()
1755 << blockArg << " to " << newArg << "\n"); in structurize()
1801 for (BlockArgument blockArg : headerBlock->getArguments()) in structurize() local
1802 mergeBlock->addArgument(blockArg.getType(), blockArg.getLoc()); in structurize()
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Transforms/ |
| H A D | AffinePromotion.cpp | 62 if (auto blockArg = coordinate.dyn_cast<mlir::BlockArgument>()) { in analyzeCoordinate() local
63 if (isa<fir::DoLoopOp>(blockArg.getOwner()->getParentOp())) in analyzeCoordinate()
223 if (auto blockArg = value.dyn_cast<mlir::BlockArgument>()) { in toAffineExpr() local
225 if (isa<fir::DoLoopOp>(blockArg.getOwner()->getParentOp()) || in toAffineExpr()
226 isa<mlir::AffineForOp>(blockArg.getOwner()->getParentOp())) in toAffineExpr()
|
| H A D | ArrayValueCopy.cpp | 367 if (auto blockArg = ro.iterArgToBlockArg(operand->get())) { in arrayMentions() local
368 int64_t arg = blockArg.getArgNumber(); in arrayMentions()
371 appendToQueue(blockArg); in arrayMentions()
|
| /llvm-project-15.0.7/mlir/include/mlir/AsmParser/ |
| H A D | AsmParserState.h | 156 void addDefinition(BlockArgument blockArg, SMLoc location);
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | Builders.h | 363 auto blockArg = val.cast<BlockArgument>(); in setInsertionPointAfterValue() local
364 setInsertionPointToStart(blockArg.getOwner()); in setInsertionPointAfterValue()
|
| /llvm-project-15.0.7/mlir/docs/Tutorials/ |
| H A D | UnderstandingTheIRStructure.md | 238 auto blockArg = operand.cast<BlockArgument>();
240 << blockArg.getArgNumber() << "\n";
|
| /llvm-project-15.0.7/flang/lib/Optimizer/Dialect/ |
| H A D | FIROps.cpp | 3478 if (auto blockArg = value.dyn_cast<mlir::BlockArgument>()) { in valueHasFirAttribute() local
3479 if (blockArg.getOwner() && blockArg.getOwner()->isEntryBlock()) in valueHasFirAttribute()
3481 blockArg.getOwner()->getParentOp())) in valueHasFirAttribute()
3482 if (funcOp.getArgAttr(blockArg.getArgNumber(), attributeName)) in valueHasFirAttribute()
|