[MLIR] Fix the name of the MaterializeVectorPass

PiperOrigin-RevId: 224536381

diff --git a/mlir/include/mlir/Transforms/Passes.h b/mlir/include/mlir/Transforms/Passes.h
index f791662ac433ae2e5ef6877229a92f82bd3a233a..4e666f7fcd2532d277f45f1a5af1cc109c14fe68 100644 (file)
--- a/mlir/include/mlir/Transforms/Passes.h
+++ b/mlir/include/mlir/Transforms/Passes.h
@@ -48,7 +48,7 @@ FunctionPass *createVectorizePass();
 FunctionPass *createVectorizerTestPass();
 
 /// Creates a pass to lower super-vectors to target-dependent HW vectors.
-FunctionPass *createMaterializeVectors();
+FunctionPass *createMaterializeVectorsPass();
 
 /// Creates a loop unrolling pass. Default option or command-line options take
 /// effect if -1 is passed as parameter.

diff --git a/mlir/lib/Transforms/MaterializeVectors.cpp b/mlir/lib/Transforms/MaterializeVectors.cpp
index b5e0f75406eeaa73543069b841b34b12b43d37ce..4c81eb9942df3734a0bf94b045860bb5a7b701bd 100644 (file)
--- a/mlir/lib/Transforms/MaterializeVectors.cpp
+++ b/mlir/lib/Transforms/MaterializeVectors.cpp
@@ -194,8 +194,8 @@ struct MaterializationState {
   DenseMap<const MLValue *, MLValue *> *substitutionsMap;
 };
 
-struct MaterializeVectors : public FunctionPass {
-  MaterializeVectors() : FunctionPass(&MaterializeVectors::passID) {}
+struct MaterializeVectorsPass : public FunctionPass {
+  MaterializeVectorsPass() : FunctionPass(&MaterializeVectorsPass::passID) {}
 
   PassResult runOnMLFunction(MLFunction *f) override;
 
@@ -207,7 +207,7 @@ struct MaterializeVectors : public FunctionPass {
 
 } // end anonymous namespace
 
-char MaterializeVectors::passID = 0;
+char MaterializeVectorsPass::passID = 0;
 
 /// Given a shape with sizes greater than 0 along all dimensions,
 /// returns the distance, in number of elements, between a slice in a dimension
@@ -641,7 +641,7 @@ static bool emitSlice(MaterializationState *state,
 ///   3. get the superVectorType for this particular terminator and the
 ///      corresponding hardware vector type (for now limited to F32)
 ///      TODO(ntv): be more general than F32.
-///   4. emit the transitive useDef set to operate on the finer grain vector
+///   4. emit the transitive useDef set to operate on the finer-grain vector
 ///      types.
 ///
 /// Notes
@@ -709,7 +709,7 @@ static bool materialize(MLFunction *f,
   return false;
 }
 
-PassResult MaterializeVectors::runOnMLFunction(MLFunction *f) {
+PassResult MaterializeVectorsPass::runOnMLFunction(MLFunction *f) {
   using matcher::Op;
   LLVM_DEBUG(dbgs() << "\nMaterializeVectors on MLFunction\n");
   LLVM_DEBUG(f->print(dbgs()));
@@ -736,17 +736,15 @@ PassResult MaterializeVectors::runOnMLFunction(MLFunction *f) {
     terminators.insert(cast<OperationStmt>(m.first));
   }
 
-  // Call materialization.
   auto fail = materialize(f, terminators, &state);
-
   return fail ? PassResult::Failure : PassResult::Success;
 }
 
-FunctionPass *mlir::createMaterializeVectors() {
-  return new MaterializeVectors();
+FunctionPass *mlir::createMaterializeVectorsPass() {
+  return new MaterializeVectorsPass();
 }
 
-static PassRegistration<MaterializeVectors>
+static PassRegistration<MaterializeVectorsPass>
     pass("materialize-vectors", "Materializes super-vectors to vectors of the "
                                 "proper size for the hardware");