Address CI comment

llvm · Oct 21, 2024 · d46490e · d46490e
1 parent 19ba481
commit d46490e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/lib/Dialect/Torch/Transforms/Passes.cpp b/lib/Dialect/Torch/Transforms/Passes.cpp
@@ -102,10 +102,11 @@ void mlir::torch::Torch::createTorchOnnxToTorchBackendPipeline(
         Torch::createDecomposeComplexOpsPass(options.backendLegalOps));
     pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
   }
-  // TODO: Move the below two passes i.e., ScalarizeShapes and
+  // TODO: Move the combination of two passes i.e., ScalarizeShapes and
   // TorchShapeRefinementPipeline out of here and create an onnx shape
   // refinement pipeline which runs iteratively over the IR.
   // This pass scalarizes the tensor shape computations.
+  createTorchShapeRefinementPipeline(pm, options);
   pm.addNestedPass<mlir::func::FuncOp>(
       mlir::torch::Torch::createScalarizeShapesPass());
   createTorchShapeRefinementPipeline(pm, options);