biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

Pruning kernel Core Tests (#26727)

Browse Source 
Signed-off-by: Fardin Hoque <kfhfar@amazon.com>
This commit is contained in:
Fardin Hoque
2025-10-13 16:08:18 -07:00
committed by GitHub
parent e3fdb627d9
commit fa96fb9c70
4 changed files with 5 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/kernels/core/test_pos_encoding.py
View File

@@ -12,8 +12,8 @@ from vllm.model_executor.layers.rotary_embedding import get_rope
from vllm.platforms import current_platform
IS_NEOX_STYLE = [True, False]
DTYPES = [torch.half, torch.bfloat16, torch.float]
HEAD_SIZES = [64, 80, 112, 120, 256]
DTYPES = [torch.bfloat16, torch.float]
HEAD_SIZES = [64, 80, 120, 256]
ROTARY_DIMS = [None, 32] # None means rotary dim == head size
NUM_HEADS = [17] # Arbitrary values for testing
BATCH_SIZES = [5] # Arbitrary values for testing