diff options
Diffstat (limited to 'mlir/lib/Dialect/GPU/Transforms/ParallelLoopMapper.cpp')
| -rw-r--r-- | mlir/lib/Dialect/GPU/Transforms/ParallelLoopMapper.cpp | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/mlir/lib/Dialect/GPU/Transforms/ParallelLoopMapper.cpp b/mlir/lib/Dialect/GPU/Transforms/ParallelLoopMapper.cpp index 84f2a907cafb..72e0ebc132e8 100644 --- a/mlir/lib/Dialect/GPU/Transforms/ParallelLoopMapper.cpp +++ b/mlir/lib/Dialect/GPU/Transforms/ParallelLoopMapper.cpp @@ -11,15 +11,20 @@ // //===----------------------------------------------------------------------===// -#include "mlir/Dialect/GPU/Transforms/ParallelLoopMapper.h" +#include "mlir/Dialect/GPU/Transforms/Passes.h" -#include "PassDetail.h" +#include "mlir/Dialect/Func/IR/FuncOps.h" #include "mlir/Dialect/GPU/IR/GPUDialect.h" -#include "mlir/Dialect/GPU/Transforms/Passes.h" +#include "mlir/Dialect/GPU/Transforms/ParallelLoopMapper.h" #include "mlir/Dialect/SCF/IR/SCF.h" #include "mlir/IR/AffineMap.h" namespace mlir { +#define GEN_PASS_DEF_GPUMAPPARALLELLOOPSPASS +#include "mlir/Dialect/GPU/Transforms/Passes.h.inc" +} // namespace mlir + +namespace mlir { using scf::ParallelOp; @@ -129,7 +134,7 @@ static void mapParallelOp(ParallelOp parallelOp, namespace { struct GpuMapParallelLoopsPass - : public GpuMapParallelLoopsPassBase<GpuMapParallelLoopsPass> { + : public impl::GpuMapParallelLoopsPassBase<GpuMapParallelLoopsPass> { void runOnOperation() override { for (Region ®ion : getOperation()->getRegions()) { region.walk([](ParallelOp parallelOp) { mapParallelOp(parallelOp); }); |