[Test]add ut for deepseek_v2. (#1964)
What this PR does / why we need it?
Add uts for deepseek_v2
Does this PR introduce any user-facing change?
No
How was this patch tested?
- vLLM version: v0.9.2
- vLLM main:
f3137cdd81
---------
Signed-off-by: 张帮政 <zhangbangzheng@huawei.com>
This commit is contained in:
@@ -188,7 +188,6 @@ class TestAscendFusedMoe:
|
||||
assert layer.top_k == default_moe_config['top_k']
|
||||
assert hasattr(layer, 'w13_weight')
|
||||
assert hasattr(layer, 'w2_weight')
|
||||
assert layer.moe_instance_id == 0
|
||||
|
||||
# check group_topk
|
||||
with pytest.raises(AssertionError):
|
||||
|
||||
Reference in New Issue
Block a user