Push shardy inliner down past ShardingGroupImport pass for late inlining.

ShardingGroupImport walks on sharding groups and verifies the groupings and simplifies their group numbers and erases duplicates. The pass does not do anything specific for ShardableDataFlowOps generally neither for NamedComputationOps specifically. Inlining and not-inlining should not impact the pass.

PiperOrigin-RevId: 902624938

Author: ekayaaslan

shardy/dialect/sdy/transforms/import/import_pipeline.cc

@@ -53,15 +53,15 @@ void addImportPipeline(OpPassManager& pm, int& dumpIndex,
   pm.addPass(
       createApplyShardingConstraintsPass(ApplyShardingConstraintsPassOptions{
           options.debugShardingOrigins, options.debugPropagationEdgeSharding}));
+  // The sharding group import pass must run after applying sharding
+  // constraints. This ensures we can detect sharding conflicts between group
+  // members which have pre-propagation shardings due to sharding constraints.
+  pm.addPass(createShardingGroupImportPass());
   if (options.enableLateInlining) {
     pm.addPass(createImportFuncCallsPass());
     // Keep SymbolDCEPass after ImportFuncCallsPass.
     pm.addPass(createSymbolDCEPass());
   }
-  // The sharding group import pass must run after applying sharding
-  // constraints. This ensures we can detect sharding conflicts between group
-  // members which have pre-propagation shardings due to sharding constraints.
-  pm.addPass(createShardingGroupImportPass());
 }
 
 void addImportPipeline(OpPassManager& pm, const PropagationOptions& options) {