Woosuk Kwon 580353da93

[Bugfix] Fix precisions in Gemma 1 (#5913)

2024-06-29 03:10:21 +00:00

..

async_engine

[Hardware][AMD][CI/Build][Doc] Upgrade to ROCm 6.1, Dockerfile improvements, test fixes (#5422)

2024-06-25 15:56:15 -07:00

basic_correctness

[Kernel] Flashinfer for prefill & decode, with Cudagraph support for decode (#4628)

2024-06-28 15:28:49 -07:00

core

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

distributed

[Kernel] Flashinfer for prefill & decode, with Cudagraph support for decode (#4628)

2024-06-28 15:28:49 -07:00

engine

[CI/Test] improve robustness of test (vllm_runner) (#5357)

2024-06-08 08:59:20 +00:00

entrypoints

[CI/Build] [2/3] Reorganize entrypoints tests (#5904)

2024-06-28 07:59:18 -07:00

fp8_kv

Enable scaled FP8 (e4m3fn) KV cache on ROCm (AMD GPU) (#3290)

2024-04-03 14:15:55 -07:00

kernels

[Bugfix] Fix compute datatype for cutlass 3.x epilogues (#5931)

2024-06-28 17:10:34 +00:00

lora

[Kernel] Add punica dimensions for Granite 3b and 8b (#5930)

2024-06-29 10:48:25 +08:00

metrics

[Misc] Extend vLLM Metrics logging API (#5925)

2024-06-29 10:36:06 +08:00

model_executor

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

models

[Bugfix] Fix precisions in Gemma 1 (#5913)

2024-06-29 03:10:21 +00:00

multimodal

[Core] Registry for processing model inputs (#5214)

2024-06-28 12:09:56 +00:00

prefix_caching

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

prompts

[BugFix] Fix input positions for long context with sliding window (#2088)

2023-12-13 12:28:13 -08:00

quantization

[Misc] Update w4a16 compressed-tensors support to include w8a16 (#5794)

2024-06-25 19:23:35 +00:00

samplers

[CI/Build] Fix Args for _get_logits_warper in Sampler Test (#5922)

2024-06-27 11:43:04 -07:00

spec_decode

[Spec Decode] Introduce DraftModelRunner (#5799)

2024-06-28 09:17:51 -07:00

tensorizer_loader

[CI/Build] Simplify OpenAI server setup in tests (#5100)

2024-06-13 11:21:53 -07:00

tokenization

[Core] Add fault tolerance for RayTokenizerGroupPool (#5748)

2024-06-25 10:15:10 -07:00

tracing

[Misc] Add OpenTelemetry support (#4687)

2024-06-19 01:17:03 +09:00

worker

[Core] Refactor Worker and ModelRunner to consolidate control plane communication (#5408)

2024-06-25 20:30:03 -07:00

__init__.py

[Small] Formatter only checks lints in changed files (#1528)

2023-10-31 15:39:38 -07:00

conftest.py

[CI/Build] Refactor image test assets (#5821)

2024-06-26 01:02:34 -07: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

[Frontend] Customizable RoPE theta (#5197)

2024-06-11 10:42:26 -07:00

test_inputs.py

[Core] Consolidate prompt arguments to LLM engines (#4328)

2024-05-28 13:29:31 -07:00

test_logger.py

[mypy] Enable type checking for test directory (#5017)

2024-06-15 04:45:31 +00:00

test_logits_processor.py

[Misc] Remove unnecessary ModelRunner imports (#4703)

2024-05-09 00:17:17 -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_sequence.py

[CI/Build] Move test_utils.py to tests/utils.py (#4425)

2024-05-13 23:50:09 +09:00

test_sharded_state_loader.py

[CI] Upgrade codespell version. (#5381)

2024-06-12 10:06:14 -07:00

test_utils.py

[CI/Build] Add unit testing for FlexibleArgumentParser (#5798)

2024-06-25 12:18:03 -07:00

utils.py

[Hardware][AMD][CI/Build][Doc] Upgrade to ROCm 6.1, Dockerfile improvements, test fixes (#5422)

2024-06-25 15:56:15 -07:00