| /external/llvm-project/mlir/lib/Conversion/SCFToOpenMP/ |
| D | SCFToOpenMP.cpp | 28 LogicalResult matchAndRewrite(scf::ParallelOp parallelOp, in matchAndRewrite()
31 if (parallelOp.getNumResults() != 0) in matchAndRewrite()
33 parallelOp, in matchAndRewrite()
39 rewriter.setInsertionPointToEnd(parallelOp.getBody()); in matchAndRewrite()
40 assert(llvm::hasSingleElement(parallelOp.region()) && in matchAndRewrite()
43 parallelOp.getBody()->getTerminator(), ValueRange()); in matchAndRewrite()
48 parallelOp.getLoc(), parallelOp.lowerBound(), parallelOp.upperBound(), in matchAndRewrite()
49 parallelOp.step()); in matchAndRewrite()
50 rewriter.inlineRegionBefore(parallelOp.region(), loop.region(), in matchAndRewrite()
52 rewriter.eraseOp(parallelOp); in matchAndRewrite()
[all …]
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | ParallelLoopMapper.cpp | 122 static void mapParallelOp(ParallelOp parallelOp, in mapParallelOp() argument
125 if (parallelOp.getAttr(getMappingAttrName()) || in mapParallelOp()
126 ((mappingLevel == MapGrid) && parallelOp->getParentOfType<ParallelOp>())) in mapParallelOp()
129 MLIRContext *ctx = parallelOp.getContext(); in mapParallelOp()
132 attrs.reserve(parallelOp.getNumLoops()); in mapParallelOp()
133 for (int i = 0, e = parallelOp.getNumLoops(); i < e; ++i) { in mapParallelOp()
138 setMappingAttr(parallelOp, attrs); in mapParallelOp()
142 for (Operation &op : *parallelOp.getBody()) { in mapParallelOp()
149 region.walk([](ParallelOp parallelOp) { mapParallelOp(parallelOp); }); in greedilyMapParallelSCFToGPU() argument
|
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPU.cpp | 289 LogicalResult matchAndRewrite(ParallelOp parallelOp,
378 ParallelOp parallelOp, gpu::LaunchOp launchOp, in processParallelLoop() argument
384 parallelOp->getAttrOfType<ArrayAttr>(gpu::getMappingAttrName()); in processParallelLoop()
387 if (!mapping || parallelOp.getNumResults() != 0) in processParallelLoop()
390 Location loc = parallelOp.getLoc(); in processParallelLoop()
405 for (auto config : llvm::zip(mapping, parallelOp.getInductionVars(), in processParallelLoop()
406 parallelOp.lowerBound(), parallelOp.upperBound(), in processParallelLoop()
407 parallelOp.step())) { in processParallelLoop()
413 return parallelOp.emitOpError() in processParallelLoop()
462 parallelOp, in processParallelLoop()
[all …]
|
| /external/llvm-project/mlir/lib/Conversion/SCFToStandard/ |
| D | SCFToStandard.cpp | 200 LogicalResult matchAndRewrite(mlir::scf::ParallelOp parallelOp,
404 ParallelLowering::matchAndRewrite(ParallelOp parallelOp, in matchAndRewrite() argument
406 Location loc = parallelOp.getLoc(); in matchAndRewrite()
413 SmallVector<Value, 4> iterArgs = llvm::to_vector<4>(parallelOp.initVals()); in matchAndRewrite()
415 ivs.reserve(parallelOp.getNumLoops()); in matchAndRewrite()
419 llvm::zip(parallelOp.getInductionVars(), parallelOp.lowerBound(), in matchAndRewrite()
420 parallelOp.upperBound(), parallelOp.step())) { in matchAndRewrite()
445 yieldOperands.reserve(parallelOp.getNumResults()); in matchAndRewrite()
446 for (auto &op : *parallelOp.getBody()) { in matchAndRewrite()
460 rewriter.eraseOp(parallelOp.getBody()->getTerminator()); in matchAndRewrite()
[all …]
|
| /external/llvm-project/flang/lib/Lower/ |
| D | OpenMP.cpp | 184 auto parallelOp = firOpBuilder.create<mlir::omp::ParallelOp>( in genOMP() local
196 parallelOp.default_valAttr(firOpBuilder.getStringAttr( in genOMP()
200 parallelOp.default_valAttr( in genOMP()
205 parallelOp.default_valAttr(firOpBuilder.getStringAttr( in genOMP()
209 parallelOp.default_valAttr(firOpBuilder.getStringAttr( in genOMP()
218 parallelOp.proc_bind_valAttr( in genOMP()
223 parallelOp.proc_bind_valAttr( in genOMP()
228 parallelOp.proc_bind_valAttr( in genOMP()
235 createBodyOfOp<omp::ParallelOp>(parallelOp, firOpBuilder, currentLocation); in genOMP()
|
| D | OpenACC.cpp | 424 auto parallelOp = createRegionOp<mlir::acc::ParallelOp, mlir::acc::YieldOp>( in genACCParallelOp() local
428 parallelOp.setAttr(mlir::acc::ParallelOp::getAsyncAttrName(), in genACCParallelOp()
431 parallelOp.setAttr(mlir::acc::ParallelOp::getWaitAttrName(), in genACCParallelOp()
434 parallelOp.setAttr(mlir::acc::ParallelOp::getSelfAttrName(), in genACCParallelOp()
|
| /external/llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| D | Utils.cpp | 53 } else if (auto parallelOp = dyn_cast<AffineParallelOp>(parentOp)) { in getOutermostInvariantForOp() local
54 for (auto iv : parallelOp.getIVs()) in getOutermostInvariantForOp()
|
| /external/llvm-project/mlir/lib/Conversion/AffineToStandard/ |
| D | AffineToStandard.cpp | 387 auto parallelOp = rewriter.create<scf::ParallelOp>(loc, lowerBoundTuple, in matchAndRewrite() local
389 rewriter.eraseBlock(parallelOp.getBody()); in matchAndRewrite()
390 rewriter.inlineRegionBefore(op.region(), parallelOp.region(), in matchAndRewrite()
391 parallelOp.region().end()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/service/mlir_gpu/ |
| D | passes.cc | 92 if (auto parallelOp = in findStore() local
94 return findStore(parallelOp.getOperation(), matches); in findStore()
|