Skip to content

Commit 3108873

Browse files
committed
Drop ascend scheduler
Signed-off-by: wangxiyuan <[email protected]>
1 parent cb33b09 commit 3108873

File tree

8 files changed

+12
-751
lines changed

8 files changed

+12
-751
lines changed

tests/ut/ops/test_linear.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,6 @@ def test_oproj_tp(self):
9999

100100
ascend_config._ASCEND_CONFIG = MagicMock()
101101
ascend_config._ASCEND_CONFIG.oproj_tensor_parallel_size = 2
102-
ascend_config._ASCEND_CONFIG.ascend_scheduler_config.enabled = False
103102

104103
linear = AscendRowParallelLinear(
105104
input_size=16,

tests/ut/test_platform.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,6 @@ def mock_vllm_config():
3232
def mock_vllm_ascend_config():
3333
mock_ascend_config = MagicMock()
3434
mock_ascend_config.torchair_graph_config.enabled = False
35-
mock_ascend_config.ascend_scheduler_config.enabled = False
3635
mock_ascend_config.enable_shared_expert_dp = False
3736
return mock_ascend_config
3837

vllm_ascend/ascend_config.py

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,6 @@ def __init__(self, vllm_config):
3939
self.torchair_graph_config = TorchairGraphConfig(
4040
torchair_graph_config, vllm_config, additional_config)
4141

42-
ascend_scheduler_config = additional_config.get(
43-
"ascend_scheduler_config", {})
44-
self.ascend_scheduler_config = AscendSchedulerConfig(
45-
ascend_scheduler_config)
46-
4742
# Dump / PrecisionDebugger configuration
4843
dump_config_path = additional_config.get("dump_config", None)
4944
self.dump_config = DumpConfig(dump_config_path)
@@ -224,20 +219,6 @@ def __init__(self, torchair_graph_config, vllm_config, additional_config):
224219
)
225220

226221

227-
class AscendSchedulerConfig:
228-
"""
229-
Configuration Object for ascend_scheduler_config from additional_config
230-
"""
231-
232-
def __init__(self, ascend_scheduler_config: dict):
233-
self.enabled = ascend_scheduler_config.get("enabled", False)
234-
# Ascend scheduler is based on vllm v0 scheduler, so we should support
235-
# all vllm v0 scheduler configs as well.
236-
for k, v in ascend_scheduler_config.items():
237-
if not hasattr(self, k):
238-
setattr(self, k, v)
239-
240-
241222
class DumpConfig:
242223
"""
243224
Configuration object for dump/PrecisionDebugger settings.

vllm_ascend/core/schedule_config.py

Lines changed: 0 additions & 105 deletions
This file was deleted.

0 commit comments

Comments
 (0)