EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Remove mindie_turbo (#4896)

Browse Source 
mindie_turbo is out of data for long time. This PR remove the related register method.

- vLLM version: v0.12.0
- vLLM main:
ad32e3e19c

Signed-off-by: wangxiyuan <wangxiyuan1007@gmail.com>
This commit is contained in:
wangxiyuan
2025-12-11 18:46:12 +08:00
committed by GitHub
parent b89763f1ed
commit 06a66939cd
4 changed files with 1 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/ut/worker/test_worker_v1.py
View File

@@ -52,14 +52,12 @@ class TestNPUWorker(TestBase):
@patch("vllm_ascend.worker.worker_v1.get_ascend_config")
@patch("vllm_ascend.worker.worker_v1.init_ascend_config")
@patch("vllm_ascend.worker.worker_v1.check_ascend_device_type")
@patch("vllm_ascend.worker.worker_v1.try_register_lib")
@patch(init_cached_hf_modules_path)
@patch("vllm_ascend.worker.worker_v1.NPUWorker._init_profiler")
def test_init_npu_worker_normal_case(
self,
mock_init_profiler,
mock_init_cached_hf_modules,
mock_try_register_lib,
mock_check_ascend_device_type,
mock_init_ascend_config,
mock_get_ascend_config,
@@ -94,12 +92,6 @@ class TestNPUWorker(TestBase):
mock_init_ascend_config.assert_called_once_with(self.vllm_config_mock)
mock_check_ascend_device_type.assert_called_once()
# Verify try_register_lib call
mock_try_register_lib.assert_called_once_with(
"mindie_turbo",
"MindIE Turbo is installed. vLLM inference will be accelerated with MindIE Turbo.",
)
# Verify cache_dtype setting
self.assertEqual(worker.cache_dtype, torch.float16)
mock_init_profiler.assert_called_once()
@@ -114,14 +106,12 @@ class TestNPUWorker(TestBase):
@patch("vllm_ascend.worker.worker_v1.get_ascend_config")
@patch("vllm_ascend.worker.worker_v1.init_ascend_config")
@patch("vllm_ascend.worker.worker_v1.check_ascend_device_type")
@patch("vllm_ascend.worker.worker_v1.try_register_lib")
@patch(init_cached_hf_modules_path)
@patch("vllm_ascend.worker.worker_v1.NPUWorker._init_profiler")
def test_init_npu_worker_with_trust_remote_code(
self,
mock_init_profiler,
mock_init_cached_hf_modules,
mock_try_register_lib,
mock_check_ascend_device_type,
mock_init_ascend_config,
mock_get_ascend_config,
@@ -159,14 +149,12 @@ class TestNPUWorker(TestBase):
@patch("vllm_ascend.worker.worker_v1.get_ascend_config")
@patch("vllm_ascend.worker.worker_v1.init_ascend_config")
@patch("vllm_ascend.worker.worker_v1.check_ascend_device_type")
@patch("vllm_ascend.worker.worker_v1.try_register_lib")
@patch(init_cached_hf_modules_path)
@patch("vllm_ascend.worker.worker_v1.NPUWorker._init_profiler")
def test_init_npu_worker_with_custom_cache_dtype(
self,
mock_init_profiler,
mock_init_cached_hf_modules,
mock_try_register_lib,
mock_check_ascend_device_type,
mock_init_ascend_config,
mock_get_ascend_config,