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

[CI] Add Triton Ascend in CI (#4921)

Browse Source 
Add triton-ascend in UT and e2e

- vLLM version: v0.12.0
- vLLM main:
ad32e3e19c
---------
Signed-off-by: Meihan-chen <jcccx.cmh@gmail.com>
This commit is contained in:
meihanc
2025-12-23 12:47:35 +08:00
committed by GitHub
parent 2e010e12dd
commit 592cfb6a6f
8 changed files with 85 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -239,7 +239,9 @@ class TestNPUWorker(TestBase):
"vllm_ascend.worker.worker.NPUWorker._init_worker_distributed_environment"
)
@patch("vllm_ascend.worker.worker.NPUPlatform")
def test_init_device(self, mock_platform, mock_init_dist_env):
@patch("vllm_ascend.worker.worker.init_device_properties_triton")
def test_init_device(self, mock_init_triton, mock_platform,
mock_init_dist_env):
"""Test _init_device method"""
from vllm_ascend.worker.worker import NPUWorker