From 016337eaec377e590313d12fceae32d9bdb22e22 Mon Sep 17 00:00:00 2001 From: lilinsiman Date: Fri, 7 Nov 2025 11:35:24 +0800 Subject: [PATCH] [v0.11.0][UT] Add new ut case for aclgraph enable (#4038) ### What this PR does / why we need it? add new ut case for aclgraph enable ### Does this PR introduce _any_ user-facing change? no ### How was this patch tested? ut Signed-off-by: lilinsiman --- tests/ut/test_platform.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+) diff --git a/tests/ut/test_platform.py b/tests/ut/test_platform.py index c6a9068..ffb7018 100644 --- a/tests/ut/test_platform.py +++ b/tests/ut/test_platform.py @@ -9,6 +9,7 @@ from torch.distributed import ProcessGroup from torch.distributed.distributed_c10d import PrefixStore from vllm.config import CompilationLevel from vllm.config.compilation import CUDAGraphMode +from vllm.engine.arg_utils import EngineArgs from vllm.platforms import PlatformEnum from tests.ut.base import TestBase @@ -739,3 +740,26 @@ class TestNPUPlatform(TestBase): group_size=4, timeout=timedelta(seconds=30), ) + + def test_aclgraph_enable(self): + config = EngineArgs() + VllmConfig = config.create_engine_config() + self.assertEqual(VllmConfig.compilation_config.cudagraph_mode, + CUDAGraphMode.PIECEWISE) + + with self.assertLogs(logger="vllm", level="INFO") as cm: + from vllm_ascend import platform + + importlib.reload(platform) + self.platform.check_and_update_config(VllmConfig) + self.assertTrue( + "PIECEWISE compilation enabled on NPU. use_inductor not supported - " + "using only ACL Graph mode" in cm.output[1]) + self.assertEqual( + VllmConfig.compilation_config.level, + CompilationLevel.PIECEWISE, + ) + self.assertEqual( + VllmConfig.compilation_config.cudagraph_mode, + CUDAGraphMode.PIECEWISE, + )