AzizCode92 2c9ed5b738 chore: Improve GPU cleanup in tests ...

Centralizes the GPU memory cleanup logic into a single static method to prevent flaky test failures from OOM errors.

Signed-off-by: AzizCode92 <azizbenothman76@gmail.com>

2025-09-06 23:03:49 +02:00

..

async_engine

[Doc]: fix typos in Python comments (#24077)

2025-09-02 02:38:55 -07:00

basic_correctness

[test][RL] Add sleep level 2 test and fix reload with sleep mode (#23521)

2025-08-26 00:25:52 +08:00

benchmarks

Feature/benchmark/random mm data/images (#23119)

2025-08-25 01:28:35 -07:00

compile

[Bugfix] Fix silu_mul+quant fusion test (#24341)

2025-09-05 20:13:42 +00:00

config

[V0 Deprecation] Remove multi-step scheduling (#22138)

2025-08-12 20:18:39 -07:00

core

[Doc]: fix typos in Python comments (#24093)

2025-09-02 21:05:45 -07:00

cuda

[Bugfix][Ray] Set the cuda context eagerly in the ray worker (#19583)

2025-06-19 22:01:16 -07:00

detokenizer

[V1] support min_tokens for detokener (#22014)

2025-08-16 02:28:10 +00:00

distributed

[Feature] Support Decode Context Parallel (DCP) for MLA (#23734)

2025-09-06 13:24:05 +08:00

encoder_decoder

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

engine

[Doc]: fix typos in Python comments (#24077)

2025-09-02 02:38:55 -07:00

entrypoints

[Fix] [gpt-oss] fix non-tool calling path for chat completion (#24324)

2025-09-06 19:10:32 +00:00

evals/gsm8k

[CI/Build] Replace lm-eval gsm8k tests with faster implementation (#23002)

2025-08-19 15:07:30 -07:00

fastsafetensors_loader

[Core] Support model loader plugins (#21067)

2025-07-24 01:49:44 -07:00

kernels

[Bugfix] Fix test_mixtral_moe (#24371)

2025-09-06 09:32:03 -07:00

kv_transfer

Remove deprecated PyNcclConnector (#24151)

2025-09-03 22:49:16 +00:00

lora

[CI/Build] Reduce the number of redundant cases to test for LoRA (#24276)

2025-09-04 21:58:44 +00:00

metrics

[V0 Deprecation] Remove multi-step scheduling (#22138)

2025-08-12 20:18:39 -07:00

mistral_tool_use

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

model_executor

[V0 deprecation] Guided decoding (#21347)

2025-07-29 03:15:30 -07:00

models

[New Model]: google/embeddinggemma-300m (#24318)

2025-09-05 22:58:36 -07:00

mq_llm_engine

[Flaky CI] Increase timeout tolerance for test_mp_crash_detection+test_default_mm_lora_chat_completions (#23028)

2025-08-16 18:33:08 +00:00

multimodal

[CI/Build] Serve images used by multimodal tests through local HTTP Server (#23907)

2025-09-03 16:13:11 +08:00

neuron

[Misc] Remove deprecated args in v0.10 (#21349)

2025-07-22 05:26:39 -07:00

plugins

[docs][misc] IOProcessor plugins fixes (#24046)

2025-09-01 09:17:41 -07:00

plugins_tests

[Core][Model] Terratorch backend integration (#23513)

2025-09-04 00:22:41 -07:00

prompts

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

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

quantization

fix some typos (#24071)

2025-09-02 20:44:50 -07:00

reasoning

Add think chunk (#21333)

2025-07-23 21:51:32 -07:00

runai_model_streamer_test

[Core] Support model loader plugins (#21067)

2025-07-24 01:49:44 -07:00

samplers

fix some typos (#24071)

2025-09-02 20:44:50 -07:00

speculative_decoding/speculators

Add: SupportsEagle3 interface for explicit EAGLE3 support (#22642)

2025-08-12 09:24:52 -07:00

standalone_tests

[CI] Remove duplicated docs build from buildkite (#22924)

2025-08-15 05:58:06 -07:00

system_messages

[V1] Implement Cascade Attention (#11635)

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

tensorizer_loader

[CI/Build] Fix tensorizer test for load_format change (#22583)

2025-08-10 19:30:00 -07:00

tokenization

[Multimodal] Consolidate mm inputs into MultiModalFeatureSpec (#23779)

2025-08-29 18:36:57 +08:00

tool_use

[gpt-oss] tool parser supports for /chat/completions [1/n] (#22386)

2025-09-04 20:39:12 -07:00

tools

[Perf] Validate @config in pre-commit instead of dynamically (#20200)

2025-07-01 05:10:28 -04:00

tpu

[V0 Deprecation] Remove multi-step scheduling (#22138)

2025-08-12 20:18:39 -07:00

tracing

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

utils_

Adds json_count_leaves utility function (#23899)

2025-08-29 05:28:13 -07:00

v1

[CI] Disable flaky structured output test from CI (#24366)

2025-09-06 13:31:56 +00:00

vllm_test_utils

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

weight_loading

[Kernel/Quant] Remove the original marlin format and qqq (#23204)

2025-08-20 15:13:36 -04:00

worker

[V0 Deprecation] Remove pooling model support in V0 (#23434)

2025-08-29 00:04:02 -07:00

__init__.py

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

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

build_cython.py

[Refactor] Refactor import utils (#20269)

2025-07-01 18:05:42 -07:00

conftest.py

[CI/Build] Serve images used by multimodal tests through local HTTP Server (#23907)

2025-09-03 16:13:11 +08:00

test_cache_block_hashing.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_config.py

[Model] Pooling models default to using chunked prefill & prefix caching if supported. (#20930)

2025-08-11 09:41:37 -07:00

test_embedded_commit.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_inputs.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_logger.py

feat: Add --enable-log-outputs flag for logging model generations (#20707)

2025-08-06 23:10:13 -07:00

test_outputs.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_pooling_params.py

[Model] Pooling model activation supports per request control by PoolingParams (#20538)

2025-08-05 00:37:00 -07:00

test_regression.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_routing_simulator.py

[Misc] Support routing logic simulation (#21990)

2025-08-06 23:06:20 -07:00

test_sampling_params.py

[V0 deprecation] Guided decoding (#21347)

2025-07-29 03:15:30 -07:00

test_scalartype.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_seed_behavior.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_sequence.py

[Bugfix] fix IntermediateTensors equal method (#23027)

2025-08-18 02:58:11 -07:00

test_sharded_state_loader.py

[Misc] normalize multiprocessing Queue usage (#22371)

2025-08-08 01:57:27 +00:00

test_test.py

[Model] Gemma3n MM (#20495)

2025-08-09 09:56:25 -07:00

test_triton_utils.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_version.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_vllm_port.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

utils.py

chore: Improve GPU cleanup in tests

2025-09-06 23:03:49 +02:00