Push shardy inliner down out of shardy-import pipeline for early inliner.

It practically combines late and early inliner.

PiperOrigin-RevId: 904960233

Author: ekayaaslan

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

@@ -51,11 +51,6 @@ void addImportPipeline(OpPassManager& pm, int& dumpIndex,
   // 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());
-  }
 }
 
 void addImportPipeline(OpPassManager& pm, const PropagationOptions& options) {

shardy/dialect/sdy/transforms/import/test/import_pipeline_enable_late_inlining_false.mlir

@@ -54,11 +54,9 @@ void populateExportOptions(ExportOptions& options,
 void addPropagationPipeline(OpPassManager& pm, int& dumpIndex,
                             const PropagationOptions& options) {
   addImportPipeline(pm, dumpIndex, options);
-  if (options.enableLateInlining) {
-    pm.addPass(createImportFuncCallsPass());
-    // Keep SymbolDCEPass after ImportFuncCallsPass.
-    pm.addPass(createSymbolDCEPass());
-  }
+  pm.addPass(createImportFuncCallsPass());
+  // Keep SymbolDCEPass after ImportFuncCallsPass.
+  pm.addPass(createSymbolDCEPass());
   {
     PropagationOptions optionsWithKeepShardingRules = options;
     optionsWithKeepShardingRules.keepShardingRules = true;