Add OptimizeLinalgOpsPass as nested func pass.

PiperOrigin-RevId: 523339319
tensorflow · Apr 11, 2023 · 9a493a7 · 9a493a7
1 parent 67913b5
commit 9a493a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow/compiler/xla/service/cpu/hlo_xla_runtime_pipeline.cc b/tensorflow/compiler/xla/service/cpu/hlo_xla_runtime_pipeline.cc
@@ -181,7 +181,8 @@ static Status CreateHloXlaPipeline(
   pm.addPass(mlir::createCSEPass());
   pm.addPass(mlir::memref::createResolveShapedTypeResultDimsPass());
   pm.addPass(mlir::createCanonicalizerPass());
-  pm.addPass(mlir::gml_st::createOptimizeLinalgOpsPass());
+  pm.addNestedPass<mlir::func::FuncOp>(
+      mlir::gml_st::createOptimizeLinalgOpsPass());
   if (options.enable_tiling_and_fusion) {
     mlir::gml_st::GmlStCPUTilingOptions opts =
         mlir::gml_st::getDefaultCPUPipelineOptions(options.cpu_name);