| /llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/ |
| H A D | VectorToGPU.cpp | 414 valueMapping[op.getResult()] = load; in convertTransferReadOp()
464 valueMapping[op.getResult()] = result; in convertConstantOpMmaSync()
508 valueMapping[op] = newOp->getResult(0); in creatLdMatrixCompatibleLoads()
593 valueMapping[op.getResult()] = result; in createNonLdMatrixLoads()
677 valueMapping[op.getResult()] = matmul; in convertContractOp()
692 valueMapping[op.getResult()] = matmul; in convertContractOpToMmaSync()
711 valueMapping[op.getResult()] = matrix; in convertConstantOp()
725 valueMapping[op.getResult()] = matrix; in convertBroadcastOp()
760 if (it == valueMapping.end()) in convertForOp()
785 if (it == valueMapping.end()) in convertYieldOp()
[all …]
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | LoopPipelining.cpp | 52 DenseMap<Value, llvm::SmallVector<Value>> valueMapping; member
157 auto it = valueMapping.find(op->getOperand(opIdx)); in emitPrologue()
158 if (it != valueMapping.end()) in emitPrologue()
217 Value valueVersion = valueMapping[forOp.getRegionIterArgs()[retVal.index()]] in createKernelLoop()
228 valueMapping[value][maxStage - info.lastUseStage + stageIdx]; in createKernelLoop()
255 valueMapping.clear(); in createKernel()
393 auto it = valueMapping.find(op->getOperand(opIdx)); in emitEpilogue()
394 if (it != valueMapping.end()) { in emitEpilogue()
429 auto it = valueMapping.find(key); in setValueMapping()
432 if (it == valueMapping.end()) in setValueMapping()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Transforms/ |
| H A D | InliningUtils.h | 70 BlockAndValueMapping &valueMapping) const { in isLegalToInline() argument
82 BlockAndValueMapping &valueMapping) const { in isLegalToInline() argument
173 BlockAndValueMapping &valueMapping) const;
175 BlockAndValueMapping &valueMapping) const;
|
| /llvm-project-15.0.7/mlir/lib/Transforms/Utils/ |
| H A D | InliningUtils.cpp | 69 BlockAndValueMapping &valueMapping) const { in isLegalToInline()
71 return handler->isLegalToInline(dest, src, wouldBeCloned, valueMapping); in isLegalToInline()
77 BlockAndValueMapping &valueMapping) const { in isLegalToInline()
79 return handler->isLegalToInline(op, dest, wouldBeCloned, valueMapping); in isLegalToInline()
115 BlockAndValueMapping &valueMapping) { in isLegalToInline() argument
120 shouldCloneInlinedRegion, valueMapping)) { in isLegalToInline()
131 shouldCloneInlinedRegion, valueMapping); in isLegalToInline()
|
| /llvm-project-15.0.7/mlir/include/mlir/Target/LLVMIR/ |
| H A D | ModuleTranslation.h | 75 llvm::Value *&llvm = valueMapping[value]; in mapValue()
83 return valueMapping.lookup(value); in lookupValue()
311 DenseMap<Value, llvm::Value *> valueMapping; variable
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorDialect.cpp | 27 BlockAndValueMapping &valueMapping) const final { in isLegalToInline()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefDialect.cpp | 27 BlockAndValueMapping &valueMapping) const final { in isLegalToInline()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgDialect.cpp | 47 BlockAndValueMapping &valueMapping) const final { in isLegalToInline()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Async/Transforms/ |
| H A D | AsyncToAsyncRuntime.cpp | 291 BlockAndValueMapping valueMapping; in outlineExecuteOp() local
292 valueMapping.map(functionInputs, func.getArguments()); in outlineExecuteOp()
293 valueMapping.map(execute.body().getArguments(), unwrappedOperands); in outlineExecuteOp()
298 builder.clone(op, valueMapping); in outlineExecuteOp()
|
| /llvm-project-15.0.7/mlir/docs/ |
| H A D | Interfaces.md | 50 /// 'valueMapping' contains any remapped values from within the 'src' region.
54 BlockAndValueMapping &valueMapping) const {
64 BlockAndValueMapping &valueMapping) const final {
100 BlockAndValueMapping &valueMapping) const {
102 return handler ? handler->isLegalToInline(dest, src, valueMapping) : false;
|
| /llvm-project-15.0.7/mlir/lib/Target/LLVMIR/ |
| H A D | ModuleTranslation.cpp | 414 valueMapping.erase(arg); in forgetMapping()
417 valueMapping.erase(value); in forgetMapping()
806 valueMapping.clear(); in convertOneFunction()
|
| /llvm-project-15.0.7/mlir/docs/Tutorials/Toy/ |
| H A D | Ch-4.md | 84 BlockAndValueMapping &valueMapping) const final {
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 146 BlockAndValueMapping &valueMapping) const final { in isLegalToInline()
174 return remainsLegalAfterInline(op, src, dest, valueMapping); in isLegalToInline()
191 BlockAndValueMapping &valueMapping) const final { in isLegalToInline()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/IR/ |
| H A D | SCF.cpp | 38 BlockAndValueMapping &valueMapping) const final { in isLegalToInline()
|