Searched refs:rootForOp (Results 1 – 4 of 4) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Affine/Transforms/ |
| D | LoopTiling.cpp | 110 AffineForOp rootForOp = band[0]; in getTileSizes() local
111 (void)rootForOp; in getTileSizes()
125 rootForOp.emitWarning("memory footprint unknown: using default tile " in getTileSizes()
|
| /external/llvm-project/mlir/lib/Conversion/SCFToGPU/ |
| D | SCFToGPU.cpp | 137 void createLaunch(AffineForOp rootForOp, AffineForOp innermostForOp,
199 void AffineLoopToGpuConverter::createLaunch(AffineForOp rootForOp, in createLaunch() argument
203 OpBuilder builder(rootForOp.getOperation()); in createLaunch()
207 ? builder.create<ConstantIndexOp>(rootForOp.getLoc(), 1) in createLaunch()
219 rootForOp.getLoc(), gridSizeX, gridSizeY, gridSizeZ, blockSizeX, in createLaunch()
249 id = builder.create<MulIOp>(rootForOp.getLoc(), step, id); in createLaunch()
252 builder.create<AddIOp>(rootForOp.getLoc(), *lbArgumentIt, id); in createLaunch()
259 rootForOp.erase(); in createLaunch()
|
| /external/llvm-project/mlir/include/mlir/Transforms/ |
| D | LoopUtils.h | 167 Loops tilePerfectlyNested(scf::ForOp rootForOp, ArrayRef<Value> sizes);
|
| /external/llvm-project/mlir/lib/Transforms/Utils/ |
| D | LoopUtils.cpp | 988 SmallVectorImpl<T> &forOps, T rootForOp, in getPerfectlyNestedLoopsImpl() argument
991 forOps.push_back(rootForOp); in getPerfectlyNestedLoopsImpl()
992 Block &body = rootForOp.region().front(); in getPerfectlyNestedLoopsImpl()
996 rootForOp = dyn_cast<T>(&body.front()); in getPerfectlyNestedLoopsImpl()
997 if (!rootForOp) in getPerfectlyNestedLoopsImpl()
1786 Loops mlir::tilePerfectlyNested(scf::ForOp rootForOp, ArrayRef<Value> sizes) { in tilePerfectlyNested() argument
1791 getPerfectlyNestedLoopsImpl(forOps, rootForOp, sizes.size()); in tilePerfectlyNested()
1862 TileLoops mlir::extractFixedOuterLoops(scf::ForOp rootForOp, in extractFixedOuterLoops() argument
1868 getPerfectlyNestedLoopsImpl(forOps, rootForOp, sizes.size()); in extractFixedOuterLoops()
|