init v0.23.0

Signed-off-by: Sun Ruoxi <sunruoxi@4paradigm.com>

tests/ut/spec_decode/test_llm_base_proposer.py

@@ -0,0 +1,112 @@
+#
+# Copyright (c) 2026 Huawei Technologies Co., Ltd. All Rights Reserved.
+# Copyright 2023 The vLLM team.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# This file is a part of the vllm-ascend project.
+#
+
+from __future__ import annotations
+
+from types import SimpleNamespace
+
+import pytest
+from vllm.config import CUDAGraphMode
+
+from vllm_ascend.spec_decode.llm_base_proposer import AscendSpecDecodeBaseProposer
+
+# CUDAGraphMode values whose ``has_full_cudagraphs()`` is True: FULL plus the
+# two composite modes that mix FULL with NONE / PIECEWISE.
+FULL_CUDAGRAPH_MODES = [
+    CUDAGraphMode.FULL,
+    CUDAGraphMode.FULL_DECODE_ONLY,
+    CUDAGraphMode.FULL_AND_PIECEWISE,
+]
+
+# Modes without a full cudagraph.
+NON_FULL_CUDAGRAPH_MODES = [
+    CUDAGraphMode.NONE,
+    CUDAGraphMode.PIECEWISE,
+]
+
+
+class TestDisablePaddedDrafterBatchWithFullGraph:
+    """Guard: ``disable_padded_drafter_batch=True`` + cuda graph + any full
+    cudagraph mode must raise ``NotImplementedError``.
+    """
+
+    @staticmethod
+    def _make_proposer(
+        *,
+        disable_padded_drafter_batch: bool,
+        use_cuda_graph: bool,
+        cudagraph_mode: CUDAGraphMode,
+    ) -> AscendSpecDecodeBaseProposer:
+        """Bypass ``__init__`` and set only the three attrs the guard reads.
+
+        ``cudagraph_mode`` is a real enum value so ``has_full_cudagraphs()`` is
+        exercised, not stubbed.
+        """
+        proposer = AscendSpecDecodeBaseProposer.__new__(AscendSpecDecodeBaseProposer)
+        proposer.speculative_config = SimpleNamespace(
+            disable_padded_drafter_batch=disable_padded_drafter_batch,
+        )
+        proposer.use_cuda_graph = use_cuda_graph
+        proposer.compilation_config = SimpleNamespace(cudagraph_mode=cudagraph_mode)
+        return proposer
+
+    @pytest.mark.parametrize("cudagraph_mode", FULL_CUDAGRAPH_MODES)
+    def test_guard_raises_when_padded_drafter_batch_disabled_with_full_cudagraph(self, cudagraph_mode: CUDAGraphMode):
+        """The bad combo: disable_padded + cuda graph + any full-cudagraph mode
+        is intercepted with ``NotImplementedError``."""
+        proposer = self._make_proposer(
+            disable_padded_drafter_batch=True,
+            use_cuda_graph=True,
+            cudagraph_mode=cudagraph_mode,
+        )
+
+        with pytest.raises(NotImplementedError, match="disable_padded_drafter_batch"):
+            proposer._raise_if_padded_drafter_batch_disabled_and_full_graph_enabled()
+
+    @pytest.mark.parametrize("cudagraph_mode", NON_FULL_CUDAGRAPH_MODES)
+    def test_guard_does_not_raise_without_full_cudagraph(self, cudagraph_mode: CUDAGraphMode):
+        """NONE / PIECEWISE never trip the guard, even with disable_padded + cuda graph."""
+        proposer = self._make_proposer(
+            disable_padded_drafter_batch=True,
+            use_cuda_graph=True,
+            cudagraph_mode=cudagraph_mode,
+        )
+
+        # Must not raise.
+        proposer._raise_if_padded_drafter_batch_disabled_and_full_graph_enabled()
+
+    @pytest.mark.parametrize("cudagraph_mode", FULL_CUDAGRAPH_MODES)
+    def test_guard_does_not_raise_when_padded_drafter_batch_enabled(self, cudagraph_mode: CUDAGraphMode):
+        """Padded drafter batch on (the default) is fine with any full cudagraph."""
+        proposer = self._make_proposer(
+            disable_padded_drafter_batch=False,
+            use_cuda_graph=True,
+            cudagraph_mode=cudagraph_mode,
+        )
+
+        proposer._raise_if_padded_drafter_batch_disabled_and_full_graph_enabled()
+
+    def test_guard_does_not_raise_when_eager(self):
+        """``enforce_eager`` -> ``use_cuda_graph=False`` short-circuits the guard."""
+        proposer = self._make_proposer(
+            disable_padded_drafter_batch=True,
+            use_cuda_graph=False,
+            cudagraph_mode=CUDAGraphMode.FULL,
+        )
+
+        proposer._raise_if_padded_drafter_batch_disabled_and_full_graph_enabled()