[Rebase] Rebase to vLLM 0.30.0 (#7820)

Signed-off-by: tzhouam <tzhouam@connect.ust.hk>
Signed-off-by: Gao Han <hgaoaf@connect.ust.hk>
Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
Co-authored-by: Gao Han <hgaoaf@connect.ust.hk>
This commit is contained in:
authored and GitHub committed 2026-09-22 15:52:01 +00:00
1 parent ca14e635f5
commit 59db6a4285
59 files changed
+1032 -300

No files matched your search

@@ -178,7 +178,13 @@ class TestGlmImageAdaLayerNormZeroQuantization:
def test_accepts_quant_config_parameter(self, mocker: MockerFixture):
"""Verify the class accepts quant_config parameter."""
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
layer = GlmImageAdaLayerNormZero(
embedding_dim=512,
dim=2560,
@@ -220,7 +226,13 @@ class TestGlmImageAdaLayerNormContinuousQuantization:
def test_accepts_quant_config_parameter(self, mocker: MockerFixture):
"""Verify the class accepts quant_config parameter."""
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
layer = GlmImageAdaLayerNormContinuous(
embedding_dim=2560,
conditioning_embedding_dim=512,
@@ -260,7 +272,13 @@ class TestGlmImageAttentionQuantization:
def test_accepts_quant_config_parameter(self, mocker: MockerFixture):
"""Verify GlmImageAttention accepts quant_config parameter."""
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
attn = GlmImageAttention(
dim=2560,
num_heads=64,
@@ -286,7 +304,13 @@ class TestColumnParallelModulesQuantization:
def test_column_parallel_gelu_accepts_quant_config(self, mocker: MockerFixture):
"""Verify ColumnParallelGELU accepts quant_config."""
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
layer = ColumnParallelGELU(
dim_in=2560,
dim_out=10240,
@@ -297,7 +321,13 @@ class TestColumnParallelModulesQuantization:
def test_column_parallel_silu_accepts_quant_config(self, mocker: MockerFixture):
"""Verify ColumnParallelSiLU accepts quant_config."""
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
layer = ColumnParallelSiLU(
dim_in=2560,
dim_out=10240,
@@ -312,7 +342,13 @@ class TestGlmImageFeedForwardQuantization:
def test_accepts_quant_config_parameter(self, mocker: MockerFixture):
"""Verify GlmImageFeedForward accepts quant_config parameter."""
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
ff = GlmImageFeedForward(
dim=2560,
dim_out=2560,
@@ -358,7 +394,13 @@ class TestGlmImageTransformerBlockQuantization:
def test_accepts_quant_config_parameter(self, mocker: MockerFixture):
"""Verify GlmImageTransformerBlock accepts quant_config parameter."""
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
parallel_config = DiffusionParallelConfig(
tensor_parallel_size=1,
sequence_parallel_size=1,
@@ -534,7 +576,13 @@ class TestGlmImageTransformer2DModelQuantization:
"""Verify the model accepts quant_config parameter."""
from vllm_omni.diffusion.data import OmniDiffusionConfig
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
parallel_config = DiffusionParallelConfig(
tensor_parallel_size=1,
sequence_parallel_size=1,
@@ -607,7 +655,13 @@ class TestGlmImageTransformer2DModelQuantization:
"""Verify norm_out (output layer) does NOT use quantization to preserve precision."""
from vllm_omni.diffusion.data import OmniDiffusionConfig
# vLLM 0.30 resolves online quantization in resolve_quant_method():
# a non-None online_quantization_config whose resolve_quant_method_cls()
# returns a target raises on a checkpoint-quantized layer. A bare
# MagicMock satisfies both, so declare no online quantization -- these
# tests only assert the layer accepts and stores a quant_config.
mock_quant_config = mocker.MagicMock()
mock_quant_config.online_quantization_config = None
parallel_config = DiffusionParallelConfig(
tensor_parallel_size=1,
sequence_parallel_size=1,