xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-17 06:42:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added test_top_k_per_row to test-pipeline.yaml. (#26569)

Browse Source 
Signed-off-by: Daniel Campora <961215+dcampora@users.noreply.github.com>
This commit is contained in:
Daniel Cámpora 2025-10-10 16:48:33 +02:00 committed by GitHub
parent f4ba2061cf
commit 0e67102d93
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.buildkite/test-pipeline.yaml
View File

@ -431,8 +431,9 @@ steps:
source_file_dependencies: source_file_dependencies:
- csrc/ - csrc/
- tests/kernels/core - tests/kernels/core
- tests/kernels/test_top_k_per_row.py
commands: commands:
- pytest -v -s kernels/core - pytest -v -s kernels/core kernels/test_top_k_per_row.py
- label: Kernels Attention Test %N # 23min - label: Kernels Attention Test %N # 23min
timeout_in_minutes: 35 timeout_in_minutes: 35

1
tests/kernels/test_top_k_per_row.py
View File

@ -125,7 +125,6 @@ def test_top_k_per_row(
indices, indices,
values, values,
num_rows, num_rows,
top_k,
logits.stride(0), logits.stride(0),
logits.stride(1), logits.stride(1),
) )