From 05a8d051f60f612c46b0e61b8538eaf303bfc19d Mon Sep 17 00:00:00 2001 From: RyuKosei <70006706+RyuKosei@users.noreply.github.com> Date: Thu, 20 Jun 2024 10:29:32 +0800 Subject: [PATCH] Fix run.py run_ipex_fp16_gpu (#11361) * fix a bug on run.py * Update run.py fixed the format problem --------- Co-authored-by: sgwhat --- python/llm/dev/benchmark/all-in-one/run.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/llm/dev/benchmark/all-in-one/run.py b/python/llm/dev/benchmark/all-in-one/run.py index c523b7829fb..2036f6f9eea 100644 --- a/python/llm/dev/benchmark/all-in-one/run.py +++ b/python/llm/dev/benchmark/all-in-one/run.py @@ -698,7 +698,7 @@ def run_ipex_fp16_gpu(repo_id, print(output[0]) if i >= warm_up: result[in_out].append([model.first_cost, model.rest_cost_mean, model.encoder_time, - actual_in_len, actual_out_len, load_time]) + actual_in_len, actual_out_len, load_time, model.peak_memory]) del model torch.xpu.empty_cache() return result