[TRTLLM-9998][fix] Change trtllm-gen MoE distributed tuning strategy back to INDEPENDENT (#10036)

Signed-off-by: Yukun He <[email protected]>

Author: hyukn

tensorrt_llm/_torch/custom_ops/trtllm_gen_custom_ops.py

@@ -11,9 +11,8 @@
                                        last_positive_power_of_2,
                                        next_positive_power_of_2)
 
-from ..autotuner import (AutoTuner, ConstraintSpec, DistributedTuningStrategy,
-                         DynamicTensorSpec, OptimizationProfile, TunableRunner,
-                         TuningConfig)
+from ..autotuner import (AutoTuner, ConstraintSpec, DynamicTensorSpec,
+                         OptimizationProfile, TunableRunner, TuningConfig)
 
 
 def prepare_dummy_topk_and_hook(
@@ -346,10 +345,8 @@ def get_tuning_config(cls) -> TuningConfig:
         dynamic_tensor_specs = cls.get_dynamic_tensor_specs()
         constraint_specs = cls.get_constraint_specs()
 
-        tuning_config = TuningConfig(
-            dynamic_tensor_specs=dynamic_tensor_specs,
-            constraint_specs=constraint_specs,
-            distributed_tuning_strategy=DistributedTuningStrategy.PARALLEL)
+        tuning_config = TuningConfig(dynamic_tensor_specs=dynamic_tensor_specs,
+                                     constraint_specs=constraint_specs)
 
         return tuning_config
 
@@ -670,10 +667,8 @@ def get_tuning_config(cls) -> TuningConfig:
         dynamic_tensor_specs = cls.get_dynamic_tensor_specs()
         constraint_specs = cls.get_constraint_specs()
 
-        tuning_config = TuningConfig(
-            dynamic_tensor_specs=dynamic_tensor_specs,
-            constraint_specs=constraint_specs,
-            distributed_tuning_strategy=DistributedTuningStrategy.PARALLEL)
+        tuning_config = TuningConfig(dynamic_tensor_specs=dynamic_tensor_specs,
+                                     constraint_specs=constraint_specs)
 
         return tuning_config
 
@@ -971,10 +966,8 @@ def get_tuning_config(cls) -> TuningConfig:
         dynamic_tensor_specs = cls.get_dynamic_tensor_specs()
         constraint_specs = cls.get_constraint_specs()
 
-        tuning_config = TuningConfig(
-            dynamic_tensor_specs=dynamic_tensor_specs,
-            constraint_specs=constraint_specs,
-            distributed_tuning_strategy=DistributedTuningStrategy.PARALLEL)
+        tuning_config = TuningConfig(dynamic_tensor_specs=dynamic_tensor_specs,
+                                     constraint_specs=constraint_specs)
 
         return tuning_config
 
@@ -1244,10 +1237,8 @@ def get_tuning_config(cls) -> TuningConfig:
         dynamic_tensor_specs = cls.get_dynamic_tensor_specs()
         constraint_specs = cls.get_constraint_specs()
 
-        tuning_config = TuningConfig(
-            dynamic_tensor_specs=dynamic_tensor_specs,
-            constraint_specs=constraint_specs,
-            distributed_tuning_strategy=DistributedTuningStrategy.PARALLEL)
+        tuning_config = TuningConfig(dynamic_tensor_specs=dynamic_tensor_specs,
+                                     constraint_specs=constraint_specs)
 
         return tuning_config
 
@@ -1515,10 +1506,8 @@ def get_tuning_config(cls) -> TuningConfig:
         dynamic_tensor_specs = cls.get_dynamic_tensor_specs()
         constraint_specs = cls.get_constraint_specs()
 
-        tuning_config = TuningConfig(
-            dynamic_tensor_specs=dynamic_tensor_specs,
-            constraint_specs=constraint_specs,
-            distributed_tuning_strategy=DistributedTuningStrategy.PARALLEL)
+        tuning_config = TuningConfig(dynamic_tensor_specs=dynamic_tensor_specs,
+                                     constraint_specs=constraint_specs)
 
         return tuning_config
 
@@ -1775,10 +1764,8 @@ def get_tuning_config(cls) -> TuningConfig:
         dynamic_tensor_specs = cls.get_dynamic_tensor_specs()
         constraint_specs = cls.get_constraint_specs()
 
-        tuning_config = TuningConfig(
-            dynamic_tensor_specs=dynamic_tensor_specs,
-            constraint_specs=constraint_specs,
-            distributed_tuning_strategy=DistributedTuningStrategy.PARALLEL)
+        tuning_config = TuningConfig(dynamic_tensor_specs=dynamic_tensor_specs,
+                                     constraint_specs=constraint_specs)
 
         return tuning_config