DevByteAI 1f214290d6 fix(compile): apply partition wrapper when loading AOT cached functions (#31536) ...

Signed-off-by: Devbyteai <abud6673@gmail.com>
Signed-off-by: DevByteAI <161969603+devbyteai@users.noreply.github.com>
Co-authored-by: Luka Govedič <ProExpertProg@users.noreply.github.com>

2026-01-08 17:27:26 +08:00

..

__init__.py

[torch.compile] avoid Dynamo guard evaluation overhead (#7898)

2024-08-28 16:10:12 -07:00

activation_quant_fusion.py

[torch.compile] Enable silu_mul_fp8_quant fusion without custom ops enabled (#27146)

2025-10-22 00:22:39 -04:00

backends.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

base_static_graph.py

Update Optional[x] -> x | None and Union[x, y] to x | y (#26633)

2025-10-12 09:51:31 -07:00

caching.py

[Bugfix][torch2.10] Fix test_qwen2_5_vl_compilation with 2.10 RC (#30822)

2025-12-18 08:23:31 -08:00

collective_fusion.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

compiler_interface.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

counter.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

cuda_graph.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

decorators.py

fix(compile): apply partition wrapper when loading AOT cached functions (#31536)

2026-01-08 17:27:26 +08:00

fix_functionalization.py

[Bugfix] Fix flashinfer ar+norm kernel not available issue (#29960)

2025-12-03 18:50:53 +00:00

fusion_attn.py

[Attention] Remove imports from vllm/attention/__init__.py (#29342)

2025-11-26 10:53:15 -07:00

fusion.py

[Bugfix] Fix fusion for VL models (#30244)

2025-12-14 21:22:37 +08:00

fx_utils.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

inductor_pass.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

matcher_utils.py

[ROCm][FEAT] Support AITER RMSNorm quantization fusion pass (#26575)

2025-12-23 02:07:54 -08:00

monitor.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

noop_elimination.py

Do not guard during noop elimination pass (#30095)

2025-12-05 04:10:12 +00:00

partition_rules.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

pass_manager.py

[ROCm][FEAT] Support AITER RMSNorm quantization fusion pass (#26575)

2025-12-23 02:07:54 -08:00

piecewise_backend.py

[Bugfix][torch2.10] Fix test_qwen2_5_vl_compilation with 2.10 RC (#30822)

2025-12-18 08:23:31 -08:00

post_cleanup.py

Convert formatting to use ruff instead of yapf + isort (#26247)

2025-10-05 07:06:22 -07:00

qk_norm_rope_fusion.py

[Attention] Remove imports from vllm/attention/__init__.py (#29342)

2025-11-26 10:53:15 -07:00

rocm_aiter_fusion.py

[ROCm][FEAT] Support AITER RMSNorm quantization fusion pass (#26575)

2025-12-23 02:07:54 -08:00

sequence_parallelism.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

torch25_custom_graph_pass.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

vllm_inductor_pass.py

[Misc][BE] Type coverage for vllm/compilation [1/3] (#31554)

2026-01-06 20:37:51 -05:00

wrapper.py

[compile] Stop one-off setting enable_aot_compile and use context manager instead. (#30503)

2025-12-11 20:28:03 +00:00