vllm/tests
Luka Govedič 7937009a7e
[Kernel] Replaced blockReduce[...] functions with cub::BlockReduce (#7233)
Co-authored-by: Michael Goin <michael@neuralmagic.com>
2024-08-21 20:18:00 -04:00
..
async_engine [BugFix] Avoid premature async generator exit and raise all exception variations (#7698) 2024-08-21 11:45:55 -04:00
basic_correctness [Kernel] Replaced blockReduce[...] functions with cub::BlockReduce (#7233) 2024-08-21 20:18:00 -04:00
compile register custom op for flash attn and use from torch.ops (#7536) 2024-08-15 22:38:56 -07:00
core [MISC] Add prefix cache hit rate to metrics (#7606) 2024-08-19 11:52:07 -07:00
distributed [Model] Add UltravoxModel and UltravoxConfig (#7615) 2024-08-21 22:49:39 +00:00
engine [VLM][Core] Support profiling with multiple multi-modal inputs per prompt (#7126) 2024-08-14 17:55:42 +00:00
entrypoints [Model] Add UltravoxModel and UltravoxConfig (#7615) 2024-08-21 22:49:39 +00:00
fp8_kv Enable scaled FP8 (e4m3fn) KV cache on ROCm (AMD GPU) (#3290) 2024-04-03 14:15:55 -07:00
kernels [BUG] fix crash on flashinfer backend with cudagraph disabled, when attention group_size not in [1,2,4,8] (#7509) 2024-08-21 08:54:31 -07:00
lora [Misc/Testing] Use torch.testing.assert_close (#7324) 2024-08-16 04:24:04 +00:00
metrics [Bugfix] StatLoggers: cache spec decode metrics when they get collected. (#6645) 2024-07-23 23:05:05 +00:00
model_executor [CI/Build] Move test_utils.py to tests/utils.py (#4425) 2024-05-13 23:50:09 +09:00
models [Model] Add UltravoxModel and UltravoxConfig (#7615) 2024-08-21 22:49:39 +00:00
multi_step [core] Multi Step Scheduling (#7000) 2024-08-19 13:52:13 -07:00
multimodal [VLM] Refactor MultiModalConfig initialization and profiling (#7530) 2024-08-17 13:30:55 -07:00
plugins/vllm_add_dummy_model [misc][plugin] add plugin system implementation (#7426) 2024-08-13 16:24:17 -07:00
prefix_caching [MISC] Add prefix cache hit rate to metrics (#7606) 2024-08-19 11:52:07 -07:00
prompt_adapter [CORE] Adding support for insertion of soft-tuned prompts (#4645) 2024-07-09 13:26:36 -07:00
prompts [BugFix] Fix input positions for long context with sliding window (#2088) 2023-12-13 12:28:13 -08:00
quantization [ci][test] adjust max wait time for cpu offloading test (#7709) 2024-08-20 17:12:44 -07:00
samplers [Core] Use flashinfer sampling kernel when available (#7137) 2024-08-19 03:24:03 +00:00
spec_decode [Speculative Decoding] Fixing hidden states handling in batch expansion (#7508) 2024-08-19 17:58:14 -07:00
tensorizer_loader [mypy] Misc. typing improvements (#7417) 2024-08-13 09:20:20 +08:00
tokenization [Core] Allow specifying custom Executor (#6557) 2024-07-20 01:25:06 +00:00
tracing [Core] Fix tracking of model forward time in case of PP>1 (#7440) 2024-08-16 13:46:01 -07:00
weight_loading [Kernel] Expand MoE weight loading + Add Fused Marlin MoE Kernel (#7527) 2024-08-21 16:17:10 -07:00
worker [Core] Add AttentionState abstraction (#7663) 2024-08-20 18:50:45 +00:00
__init__.py [Small] Formatter only checks lints in changed files (#1528) 2023-10-31 15:39:38 -07:00
conftest.py [Model] Add UltravoxModel and UltravoxConfig (#7615) 2024-08-21 22:49:39 +00:00
test_cache_block_hashing.py [mypy] Enable type checking for test directory (#5017) 2024-06-15 04:45:31 +00:00
test_config.py [Bugfix] Bump transformers to 4.43.2 (#6752) 2024-07-24 13:22:16 -07:00
test_embedded_commit.py [Misc] Add generated git commit hash as vllm.__commit__ (#6386) 2024-07-12 22:52:15 +00:00
test_inputs.py [Core] Support serving encoder/decoder models (#7258) 2024-08-09 10:39:41 +08:00
test_logger.py [ci][test] fix engine/logger test (#7621) 2024-08-16 23:00:59 -07:00
test_logits_processor.py [Core] Optimize SPMD architecture with delta + serialization optimization (#7109) 2024-08-18 17:57:20 -07:00
test_regression.py Bugfix: fix broken of download models from modelscope (#5233) 2024-06-06 09:28:10 -07:00
test_sampling_params.py [Bugfix] fix crash if max_tokens=None (#2570) 2024-01-23 22:38:55 -08:00
test_scalartype.py [Misc] Disambiguate quantized types via a new ScalarType (#6396) 2024-08-02 13:51:58 -07:00
test_sequence.py [Core] Optimize SPMD architecture with delta + serialization optimization (#7109) 2024-08-18 17:57:20 -07:00
test_sharded_state_loader.py [CI] Upgrade codespell version. (#5381) 2024-06-12 10:06:14 -07:00
test_utils.py [mypy] Misc. typing improvements (#7417) 2024-08-13 09:20:20 +08:00
utils.py [ci][test] adjust max wait time for cpu offloading test (#7709) 2024-08-20 17:12:44 -07:00