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

cleanup invalid prints (#18050)

Browse Source 
Signed-off-by: calvin chen <120380290@qq.com>
This commit is contained in:
Calvin Chen 2025-05-13 14:01:57 +08:00 committed by GitHub
parent dc1a821768
commit 48545728d8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/worker/hpu_model_runner.py
View File

@ -1554,10 +1554,8 @@ class HPUModelRunnerBase(ModelRunnerBase[TModelInputForHPU]):
'Please update Gaudi Software Suite.') 'Please update Gaudi Software Suite.')
with compile_only_mode_context( with compile_only_mode_context(
) if can_use_compile_only_mode else contextlib.nullcontext(): ) if can_use_compile_only_mode else contextlib.nullcontext():
print("aa")
self.warmup_all_buckets(self.bucketing_ctx.prompt_buckets, True, self.warmup_all_buckets(self.bucketing_ctx.prompt_buckets, True,
kv_caches) kv_caches)
print("bb")
self.warmup_all_buckets(self.bucketing_ctx.decode_buckets, False, self.warmup_all_buckets(self.bucketing_ctx.decode_buckets, False,
kv_caches) kv_caches)