diff --git a/mlir/include/mlir/Dialect/Affine/Passes.h b/mlir/include/mlir/Dialect/Affine/Passes.h index e152101236dc7..bc29d04287ac4 100644 --- a/mlir/include/mlir/Dialect/Affine/Passes.h +++ b/mlir/include/mlir/Dialect/Affine/Passes.h @@ -44,10 +44,6 @@ createSimplifyAffineStructuresPass(); std::unique_ptr> createAffineLoopInvariantCodeMotionPass(); -/// Creates a pass to convert all parallel affine.for's into 1-d affine.parallel -/// ops. -std::unique_ptr> createAffineParallelizePass(); - /// Apply normalization transformations to affine loop-like ops. If /// `promoteSingleIter` is true, single iteration loops are promoted (i.e., the /// loop is replaced by its loop body). diff --git a/mlir/include/mlir/Dialect/Affine/Passes.td b/mlir/include/mlir/Dialect/Affine/Passes.td index 77073aa29da73..d7c7897c65730 100644 --- a/mlir/include/mlir/Dialect/Affine/Passes.td +++ b/mlir/include/mlir/Dialect/Affine/Passes.td @@ -370,7 +370,6 @@ def AffineVectorize : Pass<"affine-super-vectorize", "func::FuncOp"> { def AffineParallelize : Pass<"affine-parallelize", "func::FuncOp"> { let summary = "Convert affine.for ops into 1-D affine.parallel"; - let constructor = "mlir::affine::createAffineParallelizePass()"; let options = [ Option<"maxNested", "max-nested", "unsigned", /*default=*/"-1u", "Maximum number of nested parallel loops to produce. " diff --git a/mlir/lib/Dialect/Affine/Transforms/AffineParallelize.cpp b/mlir/lib/Dialect/Affine/Transforms/AffineParallelize.cpp index 64f2bc6e745c3..fa0676b206826 100644 --- a/mlir/lib/Dialect/Affine/Transforms/AffineParallelize.cpp +++ b/mlir/lib/Dialect/Affine/Transforms/AffineParallelize.cpp @@ -42,6 +42,8 @@ namespace { /// Convert all parallel affine.for op into 1-D affine.parallel op. struct AffineParallelize : public affine::impl::AffineParallelizeBase { + using AffineParallelizeBase::AffineParallelizeBase; + void runOnOperation() override; }; @@ -90,8 +92,3 @@ void AffineParallelize::runOnOperation() { } } } - -std::unique_ptr> -mlir::affine::createAffineParallelizePass() { - return std::make_unique(); -}