xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 06:15:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] Flush TunableOp results before worker processes are destroyed. (#13623)

Browse Source 
Signed-off-by: Nichols A. Romero <nick.romero@amd.com>
This commit is contained in:
Nichols A. Romero 2025-02-25 05:08:20 -06:00 committed by GitHub
parent 75e9d49796
commit fa82074167
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
vllm/executor/multiproc_worker_utils.py
View File

@ -250,6 +250,15 @@ def _run_worker_process(
except Exception:
logger.exception("Worker failed")
# Flush TunableOp results when TunableOp is enabled and
# online (in situ) tuning is enabled.
# Offline tuning API (record_untuned_is_enabled()) only
# available in PyTorch 2.6 or later.
import torch.cuda.tunable as tunable
if (tunable.is_enabled() and tunable.tuning_is_enabled()
and not tunable.record_untuned_is_enabled()):
tunable.write_file()
logger.info("Worker exiting")