Searched refs:rootForOp (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/ |
| H A D | SCFToGPU.cpp | 156 void createLaunch(AffineForOp rootForOp, AffineForOp innermostForOp,
218 void AffineLoopToGpuConverter::createLaunch(AffineForOp rootForOp, in createLaunch() argument
222 OpBuilder builder(rootForOp.getOperation()); in createLaunch()
227 ? builder.create<arith::ConstantIndexOp>(rootForOp.getLoc(), 1) in createLaunch()
239 rootForOp.getLoc(), gridSizeX, gridSizeY, gridSizeZ, blockSizeX, in createLaunch()
269 id = builder.create<arith::MulIOp>(rootForOp.getLoc(), step, id); in createLaunch()
272 builder.create<arith::AddIOp>(rootForOp.getLoc(), *lbArgumentIt, id); in createLaunch()
279 rootForOp.erase(); in createLaunch()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/ |
| H A 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()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 821 SmallVectorImpl<T> &forOps, T rootForOp, in getPerfectlyNestedLoopsImpl() argument
824 forOps.push_back(rootForOp); in getPerfectlyNestedLoopsImpl()
825 Block &body = rootForOp.getRegion().front(); in getPerfectlyNestedLoopsImpl()
829 rootForOp = dyn_cast<T>(&body.front()); in getPerfectlyNestedLoopsImpl()
830 if (!rootForOp) in getPerfectlyNestedLoopsImpl()
908 Loops mlir::tilePerfectlyNested(scf::ForOp rootForOp, ArrayRef<Value> sizes) { in tilePerfectlyNested() argument
913 getPerfectlyNestedLoopsImpl(forOps, rootForOp, sizes.size()); in tilePerfectlyNested()
925 TileLoops mlir::extractFixedOuterLoops(scf::ForOp rootForOp, in extractFixedOuterLoops() argument
931 getPerfectlyNestedLoopsImpl(forOps, rootForOp, sizes.size()); in extractFixedOuterLoops()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/Utils/ |
| H A D | Utils.h | 166 Loops tilePerfectlyNested(scf::ForOp rootForOp, ArrayRef<Value> sizes);
|