ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Add traceback error output for win igpu test api in benchmark (#9607)

Browse source
This commit is contained in:
Yuwen Hu 2023-12-06 14:35:16 +08:00 committed by GitHub
parent aeb77b2ab1
commit 0e8f4020e5
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/dev/benchmark/all-in-one/run.py
View file

@ -19,6 +19,7 @@
import torch import torch
import time import time
import gc import gc
import traceback
import numpy as np import numpy as np
from datetime import date from datetime import date
@ -733,6 +734,7 @@ def run_transformer_int4_gpu_win(repo_id,
actual_in_len, actual_out_len, gpu_peak_mem]) actual_in_len, actual_out_len, gpu_peak_mem])
torch.xpu.empty_cache() torch.xpu.empty_cache()
except RuntimeError: except RuntimeError:
traceback.print_exc()
pass pass
model.to('cpu') model.to('cpu')
torch.xpu.synchronize() torch.xpu.synchronize()