iHore/vLLM
8
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
14,386 Commits 3 Branches 0 Tags
fa6a6be51978bd4b49ba0da17039e60f96dc5b13
Commit Graph
52 Commits
This Branch
This Branch
All Branches
Author SHA1 Message Date
Richard Zou d1a6e96d9e [torch.compile] Improve cold and warm start compile tests (#35709) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-03-02 19:27:06 +00:00
Richard Zou e82fbeec7b [torch.compile] Undo the fast_moe_cold_start hack in torch>=2.11 (#35475) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-03-01 21:44:22 +00:00
Richard Zou 7967e854da [BugFix] Fix sp tests (#34716) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-17 17:07:56 +00:00
Richard Zou 87789c8364 [Misc] vLLM's --enforce-eager should turn off compile and cudagraphs only (#34523) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-13 09:52:20 -08:00
Richard Zou e30cedd44b [torch.compile] Stop doing unnecessary FakeTensorProp in PiecewiseCompileInterpreter (#34093) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-10 19:15:40 -08:00
Richard Zou 341eed3d30 [torch.compile] Disable recursive pre_grad_passes (#34092) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-10 18:02:31 -05:00
Richard Zou 4df841fe75 [torch.compile] Add an option to force-enable the MOE cold start optimization (#33735) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-08 18:42:56 +00:00
Richard Zou 81fe69cae5 [torch.compile] Stop compiling identical artifacts (#34003) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-07 05:24:48 -08:00
Richard Zou 9f14c9224d Revert "[torch.compile] Significantly speed up cold start times" (#33820) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-04 21:59:59 +00:00
Richard Zou b1bb18de8d [torch.compile] Significantly speed up cold start times (#33641) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-03 09:12:11 -08:00
Richard Zou fd9c83d0e0 [torch.compile] Document the workaround to standalone_compile failing (#33571) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-02-03 07:16:55 +00:00
Richard ZouandRobert Shaw 5eac9a1b34 [torch.compile] Don't do the fast moe cold start optimization if there is speculative decoding (#33624) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
Co-authored-by: Robert Shaw <114415538+robertgshaw2-redhat@users.noreply.github.com>
 2026-02-03 03:38:49 +00:00
Richard Zou d9aa39a3bb [torch.compile] Speed up MOE handling in forward_context (#33184) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-01-27 15:17:54 -08:00
Richard Zou 3b8f0fe59e [torch.compile] Stop assuming 32 bit indexing (#33113) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-01-27 04:25:02 +00:00
Richard Zou 654a71fc3c [torch.compile] Improve Cold Start for MoEs (#32805) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-01-22 10:44:40 -05:00
Richard Zou bd292be0c0 [BugFix] Python file source reading can fail on UnicodeDecodeError (#32416) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-01-15 20:01:41 +00:00
Richard Zou a79079feef [BugFix] Fix flakiness in test_eagle_dp for PyTorch 2.10 (#31915) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2026-01-08 04:04:58 +00:00
Richard Zou 254f6b9867 [Bugfix] Fix eagle dp tests on A100 (#31241) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-12-25 00:05:04 +00:00
Richard Zou 65ac8d8dc4 [Docs] Add guide to debugging vLLM-torch.compile integration (#28094) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-11-05 21:31:46 +00:00
Richard Zou cd390b609d [compile] Turn standalone_compile back on (#27460) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-10-24 16:30:27 +00:00
Richard Zou 9b6504c307 [BugFix] Work around graph partition x torch.compile cache issue (#26956) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-10-15 20:06:11 -07:00
Richard ZouandHarry Mellor e1279ef00f [Docs] Update instructions for how to using existing torch binary (#24892) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
Co-authored-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
 2025-09-16 02:25:50 +00:00
Richard Zou 8026a335a1 [BugFix] Update AttnFusionPass cache key (#21947) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-08-01 07:11:29 -07:00
Richard Zou 04e38500ee [Bugfix] VLLM_V1 supports passing other compilation levels (#19340) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-07-29 09:35:58 -04:00
Richard Zou b2eb2b5ad7 [Kernel] Apply torch.Tag.needs_fixed_stride_order only for torch==2.6.0 (#19346) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-07-18 14:10:21 -04:00
Richard Zou 21274ab476 [CI] Update CODEOWNERS for vllm/compilation (#21185) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-07-18 06:51:12 -07:00
Richard Zou ba8c300018 [BugFix] VLLM_DISABLE_COMPILE_CACHE=1 should disable all reads and writes from the cache (#20942) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-07-15 01:26:18 +00:00
Richard Zou 667624659b [CI] cc folks on changes to vllm/compilation (#20925) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-07-14 07:52:17 -07:00
Richard Zou a3a5a47e48 [Bugfix] Fix torch.compile x LoRA for PyTorch 2.8 (#20823) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-07-11 23:06:04 -07:00
Richard Zou ed33349738 [BugFix] Fix use_cudagraph=False (#19612) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-06-19 08:23:12 +08:00
Richard Zou 91b2c17a55 [CI/Build] Fix torch nightly CI dependencies part 2 (#19589) 2025-06-15 20:01:10 +08:00
Richard Zou 42f52cc95b [CI/Build] Fix torch nightly CI dependencies (#19505) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-06-11 14:40:42 -07:00
Richard Zou 77f0d465d0 [BugFix] Allow use_cudagraph to work with dynamic VLLM_USE_V1 (#19390) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-06-11 07:54:41 +08:00
Richard Zou 3a4d417707 [Misc] Cleanup compilation tests (#19343) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-06-09 15:05:44 +08:00
Richard Zou 3d64d366e0 [Misc] Change tests/compile to use VLLM_V1 by default (#19302) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-06-08 16:06:48 +08:00
Richard Zou eaa2e51088 [Bugfix] Re-enable use_cudagraph in vLLM v1 (#19299) 
Signed-off-by: Richard Zou <zou3519@gmail.com>
 2025-06-08 08:56:12 +08:00
Richard Zou da511d54d8 Fix CompilationConfig repr (#19091) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-06-06 16:23:35 +08:00
Richard Zou 84ec470fca Improve "failed to get the hash of the compiled graph" error (#18956) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-05-30 15:00:54 +00:00
Richard Zou a521ef06e5 Use standalone_compile by default in torch >= 2.8.0 (#18846) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-05-30 06:41:58 +08:00
Richard Zou 26b4fa45be Add ability to use CUDAGraphs with use_inductor=False (#17345) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-05-29 10:16:52 +08:00
Richard Zou aa42561e40 Fix PiecewiseCompileInterpreter (#17338) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-05-28 08:40:53 +00:00
Richard Zou ea2236bf95 Add option to use torch._inductor.standalone_compile (#17057) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-05-09 12:59:04 -07:00
Richard Zou edbf2d609e [easy] Fix logspam on PiecewiseBackend errors (#17138) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-05-05 23:46:11 -07:00
Richard Zou b90b0852e9 [easy] Print number of needed GPUs in skip message (#17594) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-05-02 15:27:43 -07:00
Richard Zou 165cb56329 Ignore '<string>' filepath (#17330) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-04-28 19:23:29 -07:00
Richard Zou 7f58fb9718 Add assertion for no objects while hashing hf_config (#16930) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-04-22 09:32:22 -07:00
Richard Zou fe742aef5a [easy] Pass compile_fx only the config patches (#16845) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-04-20 12:25:19 +08:00
Richard Zou 682e0b6d2f Log how much time loading a compiled artifact takes (#16848) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-04-19 16:50:46 +00:00
Richard Zou 966c742ed2 Disable remote caching when calling compile_fx (#16611) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-04-15 22:18:28 -07:00
Richard Zou b590adfdc1 Fix vLLM x torch.compile config caching (#16491) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-04-14 23:11:11 -07:00
Richard Zou 70de35a881 Fix erroneous "model doesn't support compile" warning (#16486) 
Signed-off-by: rzou <zou3519@gmail.com>
 2025-04-11 16:24:36 +00:00
Richard Zouandyoukaichao 6278bc829e Fix incorrect filenames in vllm_compile_cache.py (#15494) 
Signed-off-by: <zou3519@gmail.com>
Signed-off-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: youkaichao <youkaichao@gmail.com>
 2025-03-27 18:33:41 +08:00