Home
last modified time | relevance | path

Searched refs:BlockArgument (Results 1 – 25 of 113) sorted by relevance

12345

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DDetensorize.cpp110 for (BlockArgument blockArgument : block.getArguments()) { in matchAndRewrite()
132 const DenseSet<BlockArgument> blockArgsToDetensor;
213 const DenseSet<BlockArgument> &blockArgsToDetensor) { in computeBranchOpDetensoring()
255 DenseSet<BlockArgument> &blockArgsToDetensor) override { in compute()
317 if (currentItem.dyn_cast<BlockArgument>()) { in compute()
318 BlockArgument currentItemBlockArgument = in compute()
319 currentItem.cast<BlockArgument>(); in compute()
408 DenseSet<BlockArgument> blockArgsToRemove; in compute()
458 for (BlockArgument blockArgument : block.getArguments()) in compute()
470 DenseSet<BlockArgument> blockArgsToDetensor; in runOnOperation()
[all …]
H A DFusionOnTensors.cpp220 SmallVector<BlockArgument> TileLoopNest::getTiedBBArgs(BlockArgument bbArg) { in getTiedBBArgs()
222 SmallVector<BlockArgument> bbArgs; in getTiedBBArgs()
230 bbArg = iterArg->get().dyn_cast<BlockArgument>(); in getTiedBBArgs()
237 OpOperand *TileLoopNest::getTiedIterArg(BlockArgument bbArg) { in getTiedIterArg()
239 SmallVector<BlockArgument> bbArgs = getTiedBBArgs(bbArg); in getTiedIterArg()
245 bool TileLoopNest::hasOtherUses(BlockArgument bbArg, in hasOtherUses()
269 SmallVector<BlockArgument> bbArgs = getTiedBBArgs(bbArg); in hasOtherUses()
270 return !all_of(bbArgs, [&](BlockArgument bbArg) { in hasOtherUses()
355 BlockArgument bbArg = consumerOp.getTiedBlockArgument(consumerOpOperand); in fuseProducer()
362 if (auto bbArg = sliceOp.getSource().dyn_cast<BlockArgument>()) { in fuseProducer()
H A DHoisting.cpp164 BlockArgument tensorArg) { in tensorChunkAccessedByUnknownOp()
243 auto bbArg = insertSliceOp.getDest().dyn_cast<BlockArgument>(); in getLoopInvariantTransferWriteOpDefining()
262 BlockArgument tensorBBArg) { in hoistReadWrite()
309 ArrayRef<BlockArgument> newBBArgs) { in hoistReadWrite()
501 ArrayRef<BlockArgument> newBBArgs) { in hoistRedundantVectorTransfers()
/llvm-project-15.0.7/mlir/lib/IR/
H A DBlock.cpp22 for (BlockArgument arg : arguments) in ~Block()
141 BlockArgument Block::addArgument(Type type, Location loc) { in addArgument()
142 BlockArgument arg = BlockArgument::create(type, this, arguments.size(), loc); in addArgument()
160 BlockArgument Block::insertArgument(unsigned index, Type type, Location loc) { in insertArgument()
163 auto arg = BlockArgument::create(type, this, index, loc); in insertArgument()
168 for (BlockArgument arg : llvm::drop_begin(arguments, index)) in insertArgument()
175 BlockArgument Block::insertArgument(args_iterator it, Type type, Location loc) { in insertArgument()
185 for (BlockArgument arg : llvm::drop_begin(arguments, index)) in eraseArgument()
198 [&](BlockArgument arg) { return eraseIndices.test(arg.getArgNumber()); }); in eraseArguments()
201 void Block::eraseArguments(function_ref<bool(BlockArgument)> shouldEraseFn) { in eraseArguments()
H A DValue.cpp30 return cast<BlockArgument>().getLoc(); in getLoc()
37 return cast<BlockArgument>().setLoc(loc); in setLoc()
44 return cast<BlockArgument>().getOwner()->getParent(); in getParentRegion()
51 return cast<BlockArgument>().getOwner(); in getParentBlock()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DValue.h24 class BlockArgument; variable
59 BlockArgument = 7 enumerator
277 return value->getKind() == ValueImpl::Kind::BlockArgument; in classof()
295 friend BlockArgument; variable
300 class BlockArgument : public Value {
320 static BlockArgument create(Type type, Block *owner, int64_t index, in create()
354 return value->getKind() != ValueImpl::Kind::BlockArgument; in classof()
483 static mlir::BlockArgument getEmptyKey() {
487 static mlir::BlockArgument getTombstoneKey() {
520 struct PointerLikeTypeTraits<mlir::BlockArgument>
[all …]
H A DBlock.h74 using BlockArgListType = MutableArrayRef<BlockArgument>;
91 BlockArgument addArgument(Type type, Location loc);
96 BlockArgument insertArgument(args_iterator it, Type type, Location loc);
104 BlockArgument insertArgument(unsigned index, Type type, Location loc);
117 void eraseArguments(function_ref<bool(BlockArgument)> shouldEraseFn);
120 BlockArgument getArgument(unsigned i) { return arguments[i]; } in getArgument()
359 std::vector<BlockArgument> arguments;
H A DRegion.h80 using BlockArgListType = MutableArrayRef<BlockArgument>;
98 BlockArgument addArgument(Type type, Location loc) { in addArgument()
105 BlockArgument insertArgument(args_iterator it, Type type, Location loc) { in insertArgument()
116 BlockArgument insertArgument(unsigned index, Type type, Location loc) { in insertArgument()
124 BlockArgument getArgument(unsigned i) { return getArguments()[i]; } in getArgument()
/llvm-project-15.0.7/mlir/lib/Analysis/AliasAnalysis/
H A DLocalAliasAnalysis.cpp61 firstInputIndex = inputs[0].cast<BlockArgument>().getArgNumber(); in collectUnderlyingAddressValues()
62 lastInputIndex = inputs.back().cast<BlockArgument>().getArgNumber(); in collectUnderlyingAddressValues()
134 static void collectUnderlyingAddressValues(BlockArgument arg, unsigned maxDepth, in collectUnderlyingAddressValues()
188 if (BlockArgument arg = value.dyn_cast<BlockArgument>()) in collectUnderlyingAddressValues()
217 if (BlockArgument arg = value.dyn_cast<BlockArgument>()) in getAllocEffectFor()
306 BlockArgument rhsArg = rhs.dyn_cast<BlockArgument>(); in aliasImpl()
/llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/
H A DAsyncParallelFor.cpp135 BlockArgument blockIndex();
136 BlockArgument blockSize();
137 ArrayRef<BlockArgument> tripCounts();
138 ArrayRef<BlockArgument> lowerBounds();
139 ArrayRef<BlockArgument> upperBounds();
140 ArrayRef<BlockArgument> steps();
141 ArrayRef<BlockArgument> captures();
144 ArrayRef<BlockArgument> args;
282 BlockArgument blockIndex = args.blockIndex(); in createParallelComputeFunction()
283 BlockArgument blockSize = args.blockSize(); in createParallelComputeFunction()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/Transforms/
H A DOneShotModuleBufferize.cpp125 BlockArgument bbArg) { in annotateEquivalentReturnBbArg()
156 for (BlockArgument bbArg : funcOp.getArguments()) in aliasingFuncOpBBArgsAnalysis()
174 static void annotateFuncArgAccess(func::FuncOp funcOp, BlockArgument bbArg, in annotateFuncArgAccess()
200 for (BlockArgument bbArg : funcOp.getArguments()) { in funcOpBbArgReadWriteAnalysis()
208 for (BlockArgument bbArg : funcOp.getArguments()) { in funcOpBbArgReadWriteAnalysis()
226 static void removeBufferizationAttributes(BlockArgument bbArg) { in removeBufferizationAttributes()
434 for (BlockArgument bbArg : op.getArguments()) in bufferizeModuleOp()
H A DBufferResultsToOutParams.cpp42 SmallVectorImpl<BlockArgument> &appendedEntryArgs) { in updateFuncOp()
94 ArrayRef<BlockArgument> appendedEntryArgs) { in updateReturnOps()
168 SmallVector<BlockArgument, 6> appendedEntryArgs; in promoteBufferResultsToOutParams()
H A DBufferDeallocation.cpp276 (definingBlock == parentBlock && value.isa<BlockArgument>())) { in introduceClones()
303 if (failed(value.isa<BlockArgument>() in introduceClones()
304 ? introduceBlockArgCopy(value.cast<BlockArgument>()) in introduceClones()
318 LogicalResult introduceBlockArgCopy(BlockArgument blockArg) { in introduceBlockArgCopy()
/llvm-project-15.0.7/mlir/lib/Analysis/
H A DSliceAnalysis.cpp95 } else if (auto blockArg = operand.dyn_cast<BlockArgument>()) { in getBackwardSliceImpl()
129 Operation *bbAargOwner = root.cast<BlockArgument>().getOwner()->getParentOp(); in getBackwardSlice()
220 ArrayRef<BlockArgument> iterCarriedArgs, in dependsOnCarriedVals()
270 Value mlir::matchReduction(ArrayRef<BlockArgument> iterCarriedArgs, in matchReduction()
275 BlockArgument redCarriedVal = iterCarriedArgs[redPos]; in matchReduction()
H A DLiveness.cpp54 for (BlockArgument argument : block->getArguments()) { in BlockInfoBuilder()
187 currentBlock = value.cast<BlockArgument>().getOwner(); in resolveLiveness()
269 for (BlockArgument argument : block->getArguments()) in print()
283 auto blockArg = value.cast<BlockArgument>(); in print()
407 if (isLiveIn(value) || value.isa<BlockArgument>()) in currentlyLiveValues()
/llvm-project-15.0.7/mlir/include/mlir/Analysis/
H A DSliceAnalysis.h20 class BlockArgument; variable
236 Value matchReduction(ArrayRef<BlockArgument> iterCarriedArgs, unsigned redPos,
/llvm-project-15.0.7/mlir/lib/Transforms/Utils/
H A DRegionUtils.cpp149 return wasProvenLive(value.cast<BlockArgument>()); in wasProvenLive()
151 bool wasProvenLive(BlockArgument arg) { return liveValues.count(arg); } in wasProvenLive()
157 setProvedLive(value.cast<BlockArgument>()); in setProvedLive()
159 void setProvedLive(BlockArgument arg) { in setProvedLive()
221 for (BlockArgument arg : successor->getArguments()) in propagateTerminatorLiveness()
343 [&](BlockArgument arg) { return !liveMap.wasProvenLive(arg); }); in deleteDeadness()
441 if (BlockArgument arg = value.dyn_cast<BlockArgument>()) in getOrderOf()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/
H A DUtils.h423 SmallVector<BlockArgument> getTiedBBArgs(BlockArgument bbArg);
427 OpOperand *getTiedIterArg(BlockArgument bbArg);
432 bool hasOtherUses(BlockArgument bbArg, tensor::ExtractSliceOp sliceOp);
/llvm-project-15.0.7/mlir/test/python/ir/
H A Dvalue.py52 assert BlockArgument.isinstance(func.regions[0].blocks[0].arguments[0])
56 assert not BlockArgument.isinstance(op.results[0])
/llvm-project-15.0.7/mlir/include/mlir/AsmParser/
H A DAsmParserState.h20 class BlockArgument; variable
156 void addDefinition(BlockArgument blockArg, SMLoc location);
/llvm-project-15.0.7/mlir/test/lib/Analysis/
H A DTestAliasAnalysis.cpp24 if (BlockArgument arg = value.dyn_cast<BlockArgument>()) { in printAliasOperand()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DNormalizeMemRefs.cpp182 BlockArgument oldMemRef = funcOp.getArgument(argIndex); in areMemRefsNormalizable()
341 funcOp.getArguments(), [](BlockArgument arg) { return arg.getLoc(); })); in normalizeFuncOpMemRefs()
366 BlockArgument newMemRef = funcOp.front().insertArgument( in normalizeFuncOpMemRefs()
368 BlockArgument oldMemRef = funcOp.getArgument(argIndex + 1); in normalizeFuncOpMemRefs()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DSuperVectorize.cpp702 void registerBlockArgVectorReplacement(BlockArgument replaced,
703 BlockArgument replacement);
716 void registerValueScalarReplacement(BlockArgument replaced,
717 BlockArgument replacement);
825 BlockArgument replaced, BlockArgument replacement) { in registerBlockArgVectorReplacement()
850 BlockArgument replaced, BlockArgument replacement) { in registerValueScalarReplacement()
/llvm-project-15.0.7/mlir/lib/Transforms/
H A DStripDebugInfo.cpp32 for (BlockArgument &arg : block.getArguments()) { in runOnOperation()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizableOpInterface.cpp45 return value.cast<BlockArgument>().getOwner()->getParentOp(); in getOwnerOfValue()
310 if (auto bbArg = value.dyn_cast<BlockArgument>()) { in setInsertionPointAfter()
407 if (condition(value) || value.isa<BlockArgument>()) { in findValueInReverseUseDefChain()
582 if (auto bbArg = value.dyn_cast<BlockArgument>()) in getBufferType()
691 auto bbArg = value.dyn_cast<BlockArgument>(); in isFunctionArgument()

12345