Skip to content
This repository was archived by the owner on Jun 8, 2023. It is now read-only.
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions python/conformance/conformance_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -321,7 +321,7 @@ def unlink_device_configs():
sys.exit(0)

for saved_pth in saved_pth_list:
cfg_name = saved_pth.split("_")[0]
cfg_name = saved_pth.rsplit("_", 1)[0]
cfg_func_name = saved_pth.split("::")[1].rsplit("_", 1)[0]
if not need_process_func(cfg_func_name, func_name, model_name):
continue
Expand Down Expand Up @@ -354,8 +354,8 @@ def unlink_device_configs():
if ck in tensor_para and ck in device_tensor_para:
if tensor_para[ck] in device_tensor_para[ck]:
skipped = True
if 'dtype' in device_cfg and tensor_para['dtype'] in device_cfg['dtype']:
skipped = True
if 'dtype' in device_cfg and tensor_para['dtype'] in device_cfg['dtype']:
skipped = True

tol_keys_list = ['atol', 'rtol', 'atol_half', 'rtol_half']
for key in tol_keys_list:
Expand Down