Add traceback error output for win igpu test api in benchmark (#9607)
This commit is contained in:
parent
aeb77b2ab1
commit
0e8f4020e5
1 changed files with 2 additions and 0 deletions
|
|
@ -19,6 +19,7 @@
|
|||
import torch
|
||||
import time
|
||||
import gc
|
||||
import traceback
|
||||
|
||||
import numpy as np
|
||||
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])
|
||||
torch.xpu.empty_cache()
|
||||
except RuntimeError:
|
||||
traceback.print_exc()
|
||||
pass
|
||||
model.to('cpu')
|
||||
torch.xpu.synchronize()
|
||||
|
|
|
|||
Loading…
Reference in a new issue