Liangfu Chen ddee88d0ff

[Neuron][Kernel] NKI-based flash-attention kernel with paged KV cache (#11277) ...

Signed-off-by: Liangfu Chen <liangfc@amazon.com>
Co-authored-by: Jiangfei Duan <jfduan@outlook.com>

2025-01-27 17:31:16 -08:00

..

async_engine

[Misc] Remove deprecated code (#12383)

2025-01-24 14:45:20 -05:00

basic_correctness

[FlashInfer] Upgrade to 0.2.0 (#11194)

2025-01-27 18:19:24 +00:00

compile

[FlashInfer] Upgrade to 0.2.0 (#11194)

2025-01-27 18:19:24 +00:00

core

[Core] Support reset_prefix_cache (#12284)

2025-01-22 18:52:27 +00:00

data

[Bugfix] Fix load config when using bools (#9533)

2024-10-27 13:46:41 -04:00

distributed

Support torchrun and SPMD-style offline inference (#12071)

2025-01-16 19:58:53 +08:00

encoder_decoder

[Encoder Decoder] Update Mllama to run with both FlashAttention and XFormers (#9982)

2024-11-12 10:53:57 -08:00

engine

[Core] Interface for accessing model from VllmRunner (#10353)

2025-01-20 15:00:59 +08:00

entrypoints

[V1][Metrics] Add initial Prometheus logger (#12416)

2025-01-27 12:26:28 -05:00

kernels

Update pre-commit hooks (#12475)

2025-01-27 17:23:08 -07:00

kv_transfer

Update pre-commit hooks (#12475)

2025-01-27 17:23:08 -07:00

lora

Update pre-commit hooks (#12475)

2025-01-27 17:23:08 -07:00

metrics

monitor metrics of tokens per step using cudagraph batchsizes (#11031)

2024-12-09 22:35:36 -08:00

model_executor

[Core] Interface for accessing model from VllmRunner (#10353)

2025-01-20 15:00:59 +08:00

models

Update pre-commit hooks (#12475)

2025-01-27 17:23:08 -07:00

mq_llm_engine

[Bugfix][core] replace heartbeat with pid check (#9818)

2024-10-30 09:34:07 -07:00

multi_step

[Misc] Remove deprecated code (#12383)

2025-01-24 14:45:20 -05:00

multimodal

[VLM] Simplify post-processing of replacement info (#12269)

2025-01-21 16:48:13 -08:00

neuron

[Neuron][Kernel] NKI-based flash-attention kernel with paged KV cache (#11277)

2025-01-27 17:31:16 -08:00

plugins

[Platform] Do not raise error if _Backend is not found (#12023)

2025-01-15 10:14:15 +00:00

plugins_tests

[Platform] Do not raise error if _Backend is not found (#12023)

2025-01-15 10:14:15 +00:00

prefix_caching

Prefix Cache Aware Scheduling [1/n] (#10128)

2024-11-22 21:15:55 -08: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

Update pre-commit hooks (#12475)

2025-01-27 17:23:08 -07:00

runai_model_streamer

[Core] Loading model from S3 using RunAI Model Streamer as optional loader (#10192)

2024-12-20 16:46:24 +00:00

samplers

Update pre-commit hooks (#12475)

2025-01-27 17:23:08 -07:00

spec_decode

[Feature] [Spec decode]: Enable MLPSpeculator/Medusa and prompt_logprobs with ChunkedPrefill (#10132)

2025-01-27 13:38:35 -08:00

standalone_tests

[ci][build] add tests for python only compilation (#10915)

2024-12-05 08:54:47 -08:00

system_messages

[V1] Implement Cascade Attention (#11635)

2025-01-01 21:56:46 +09:00

tensorizer_loader

[Core] Interface for accessing model from VllmRunner (#10353)

2025-01-20 15:00:59 +08:00

tokenization

[CI/Build] drop support for Python 3.8 EOL (#8464)

2024-11-06 07:11:55 +00:00

tool_use

[Model] IBM Granite 3.1 (#11307)

2024-12-19 11:27:24 +08:00

tpu

[TPU][Quantization] TPU W8A8 (#11785)

2025-01-08 19:33:29 +00:00

tracing

[Misc] Remove experimental dep from tracing.py (#12007)

2025-01-21 11:51:55 -08:00

v1

[V1][CI/Test] Do basic test for top-p & top-k sampling (#12469)

2025-01-27 09:40:04 -08:00

vllm_test_utils

[misc] improve memory profiling (#11809)

2025-01-08 06:36:03 +00:00

weight_loading

[Model] Support for fairseq2 Llama (#11442)

2025-01-19 10:40:40 -08:00

worker

[FP8][Kernel] Dynamic kv cache scaling factors computation (#11906)

2025-01-23 18:04:03 +00:00

__init__.py

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

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

conftest.py

[Core] Interface for accessing model from VllmRunner (#10353)

2025-01-20 15:00:59 +08:00

test_cache_block_hashing.py

[Core] Make encoder-decoder inputs a nested structure to be more composable (#9604)

2024-11-05 10:07:31 +08:00

test_config.py

[Misc] Split up pooling tasks (#10820)

2024-12-11 01:28:00 -08:00

test_embedded_commit.py

[CI/Build] use setuptools-scm to set __version__ (#4738)

2024-09-23 09:44:26 -07:00

test_inputs.py

[Core][Frontend] Add Support for Inference Time mm_processor_kwargs (#9131)

2024-10-08 14:12:56 +00:00

test_logger.py

Rename vllm.logging to vllm.logging_utils (#10134)

2024-11-08 20:53:24 +00:00

test_logits_processor.py

[Core] Factor out common code in SequenceData and Sequence (#8675)

2024-09-21 02:30:39 +00: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

[Bugfix] Fix support for dimension like integers and ScalarType (#9299)

2024-10-17 19:08:34 +00:00

test_sequence.py

[Core] Factor out common code in SequenceData and Sequence (#8675)

2024-09-21 02:30:39 +00:00

test_sharded_state_loader.py

[CI/Build] Replaced some models on tests for smaller ones (#9570)

2024-10-22 04:52:14 +00:00

test_utils.py

[core] further polish memory profiling (#12126)

2025-01-18 12:25:08 +08:00

utils.py

[ci]try to fix flaky multi-step tests (#11894)

2025-01-09 14:47:29 +00:00