Push shardy outliner up past ConstantOrScalarMerger pass.

ConstantOrScalarMerger combines constants and constant-likes that are sharded the same way. It is not critical for final performance but it is important to keep the graph smaller. Hence, unlike ConstantOrScalarSplitter, it does not take NamedComputations/Funcs/Calls as parts of constant expressions.

PiperOrigin-RevId: 905027384

Author: ekayaaslan

shardy/dialect/sdy/transforms/export/export_pipeline.cc

@@ -70,8 +70,8 @@ void runShardyPartitioner(OpPassManager& pm, int& dumpIndex,
 
 void addExportPipeline(OpPassManager& pm, int& dumpIndex,
                        const ExportOptions& options) {
-  pm.addNestedPass<func::FuncOp>(createConstantOrScalarMergerPass());
   pm.addPass(createExportNamedComputationsPass());
+  pm.addNestedPass<func::FuncOp>(createConstantOrScalarMergerPass());
   if (!options.avoidExportForPartitioning) {
     pm.addPass(createRemoveShardingGroupsPass());
     pm.addNestedPass<func::FuncOp>(createShardingConstraintToReshardPass());