e2237819a9
### What this PR does / why we need it?
The incorrect regular expression syntax `.*[UE4M3|ue4m3].*` actually
ignores all words containing any of the following characters: `u, e, 4,
m, 3, |`
```yaml
extend-ignore-identifiers-re = [".*Unc.*", ".*_thw",
".*UE8M0.*", ".*[UE4M3|ue4m3].*", ".*eles.*", ".*fo.*", ".*ba.*",
".*ot.*", ".*[Tt]h[rR].*"]
```
===fix===>
```yaml
extend-ignore-identifiers-re = [".*Unc.*", ".*_thw",
".*UE8M0.*", ".*(UE4M3|ue4m3]).*", ".*eles.*", ".*fo.*", ".*ba.*",
".*ot.*", ".*[Tt]h[rR].*"]
```
### Does this PR introduce _any_ user-facing change?
### How was this patch tested?
- vLLM version: v0.15.0
- vLLM main:
9562912cea
Signed-off-by: MrZ20 <2609716663@qq.com>
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
import unittest
|
|
from typing import ClassVar
|
|
|
|
import torch
|
|
|
|
from vllm_ascend.ops.fused_moe.moe_mlp import cumsum_group_list
|
|
|
|
|
|
class TestCumsumGroupList(unittest.TestCase):
|
|
glist_dict: ClassVar[dict[int, torch.Tensor]]
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.glist_dict = {
|
|
0: torch.tensor([0, 2, 3, 3]),
|
|
1: torch.tensor([0, 2, 1, 0]),
|
|
2: torch.tensor([[1, 2], [2, 1], [0, 0], [0, 0]])
|
|
}
|
|
|
|
support_combine = [(0, 0), (1, 0), (0, 1)]
|
|
unsupported_combine = [(0, 2), (2, 1), (1, 2)]
|
|
|
|
def test_cumsum_group_list_supported_conversion(self):
|
|
for src_list_type, dst_list_type in self.support_combine:
|
|
with self.subTest(src=src_list_type, dst=dst_list_type):
|
|
result = cumsum_group_list(self.glist_dict[src_list_type],
|
|
src_list_type,
|
|
dst_list_type,
|
|
expert_num=4)
|
|
self.assertTrue(
|
|
torch.equal(result, self.glist_dict[dst_list_type]))
|
|
|
|
def test_cumsum_group_list_invalid_type_valueerror(self):
|
|
with self.assertRaises(ValueError) as excinfo:
|
|
cumsum_group_list(self.glist_dict[0], 4, 0)
|
|
self.assertIn("group_list_type should be in [0, 1, 2], but received",
|
|
str(excinfo.exception))
|
|
|
|
def test_cumsum_group_list_unsupported_conversion_notimplementederror(
|
|
self):
|
|
for src_list_type, dst_list_type in self.unsupported_combine:
|
|
with self.subTest(src=src_list_type, dst=dst_list_type):
|
|
with self.assertRaises(NotImplementedError) as excinfo:
|
|
cumsum_group_list(self.glist_dict[0], src_list_type,
|
|
dst_list_type)
|
|
self.assertIn("This feature is under development.",
|
|
str(excinfo.exception))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=2)
|