Drop ascend scheduler

Signed-off-by: wangxiyuan <[email protected]>

Author: wangxiyuan

.github/workflows/vllm_ascend_test_pr_light.yaml

@@ -139,7 +139,6 @@ jobs:
             --ignore tests/ut/kv_connector/test_remote_prefill_lifecycle.py \
             --ignore tests/ut/kv_connector/test_remote_decode_lifecycle.py \
             --ignore tests/ut/kv_connector/test_llmdatadist_connector.py \
-            --ignore tests/ut/ops/test_linear.py \
             --ignore tests/ut/core/test_scheduler_dynamic_batch.py
 
       - name: Upload coverage to Codecov

tests/ut/ops/test_linear.py

@@ -99,7 +99,6 @@ def test_oproj_tp(self):
 
         ascend_config._ASCEND_CONFIG = MagicMock()
         ascend_config._ASCEND_CONFIG.oproj_tensor_parallel_size = 2
-        ascend_config._ASCEND_CONFIG.ascend_scheduler_config.enabled = False
 
         linear = AscendRowParallelLinear(
             input_size=16,

tests/ut/test_platform.py

@@ -32,7 +32,6 @@ def mock_vllm_config():
     def mock_vllm_ascend_config():
         mock_ascend_config = MagicMock()
         mock_ascend_config.torchair_graph_config.enabled = False
-        mock_ascend_config.ascend_scheduler_config.enabled = False
         mock_ascend_config.enable_shared_expert_dp = False
         return mock_ascend_config
 

vllm_ascend/ascend_config.py

@@ -45,11 +45,6 @@ def __init__(self, vllm_config):
         self.ascend_compilation_config = AscendCompilationConfig(
             **ascend_compilation_config)
 
-        ascend_scheduler_config = additional_config.get(
-            "ascend_scheduler_config", {})
-        self.ascend_scheduler_config = AscendSchedulerConfig(
-            ascend_scheduler_config)
-
         # Dump / PrecisionDebugger configuration
         dump_config_path = additional_config.get("dump_config", None)
         self.dump_config = DumpConfig(dump_config_path)
@@ -255,20 +250,6 @@ def __init__(self, torchair_graph_config, vllm_config, additional_config):
             )
 
 
-class AscendSchedulerConfig:
-    """
-    Configuration Object for ascend_scheduler_config from additional_config
-    """
-
-    def __init__(self, ascend_scheduler_config: dict):
-        self.enabled = ascend_scheduler_config.get("enabled", False)
-        # Ascend scheduler is based on vllm v0 scheduler, so we should support
-        # all vllm v0 scheduler configs as well.
-        for k, v in ascend_scheduler_config.items():
-            if not hasattr(self, k):
-                setattr(self, k, v)
-
-
 class DumpConfig:
     """
     Configuration object for dump/PrecisionDebugger settings.