LeeWenquan b72e34013f Add ut for mla (#2637) ...

### What this PR does / why we need it?
Update UT for MLA case

- vLLM version: v0.10.1.1
- vLLM main:
14b4326b94

---------

Signed-off-by: SunnyLee219 <3294305115@qq.com>

2025-09-01 14:07:57 +08:00

..

attention

Add ut for mla (#2637)

2025-09-01 14:07:57 +08:00

core

[CI] fix UT error. (#2644)

2025-08-30 12:04:01 +08:00

device_allocator

add ut for device allocator/camem and mutistream/layers (#2037)

2025-07-31 19:17:27 +08:00

distributed

[Feat]: Add custom lmhead tensor model parallel (#2309)

2025-08-29 11:41:21 +08:00

fake_weight

[CI] Add unit test framework (#1201)

2025-06-16 18:32:28 +08:00

kv_connector

[CI] fix UT error. (#2644)

2025-08-30 12:04:01 +08:00

models

[Feat]: Add custom lmhead tensor model parallel (#2309)

2025-08-29 11:41:21 +08:00

multistream

add ut for device allocator/camem and mutistream/layers (#2037)

2025-07-31 19:17:27 +08:00

ops

[6/N][refactor]delete torchair in rotary ops (#2581)

2025-09-01 09:10:15 +08:00

patch/worker/patch_common

[Misc] Remove redundant imported envs, using envs_ascend instead (#2193)

2025-08-14 09:33:39 +08:00

quantization

[main] [refactor] refactor fused_moe.py to enable token_dispatchers (#2570)

2025-08-28 10:13:35 +08:00

sample

[Perf][MTP] Optimize reject sampler in greedy situation. (#2137)

2025-08-11 17:37:49 +08:00

torchair

[5/N][refactor]add torchair rotary ops (#2559)

2025-09-01 09:09:21 +08:00

worker

[Test]Add unit test for worker_v1.py (#2547)

2025-08-26 22:00:49 +08:00

__init__.py

[2/4][Refactor] Refactor torchair utils (#1892)

2025-07-21 19:43:30 +08:00

base.py

[Feature]: implement the fusion of allreduce and matmul in prefill phase when tp is enabled (#1926)

2025-07-28 15:13:37 +08:00

conftest.py

[1/N][CustomOp] Register activation customop instead of overwrite forward_oot (#1841)

2025-07-18 23:07:14 +08:00

test_ascend_config.py

[Feat]: Add custom lmhead tensor model parallel (#2309)

2025-08-29 11:41:21 +08:00

test_envs.py

[Misc] Remove redundant imported envs, using envs_ascend instead (#2193)

2025-08-14 09:33:39 +08:00

test_platform.py

[CI] fix UT error. (#2644)

2025-08-30 12:04:01 +08:00

test_utils.py

[Feat]: Add custom lmhead tensor model parallel (#2309)

2025-08-29 11:41:21 +08:00