From 3dda56e84a37b6c398c5fa0c24632b72e37cdec8 Mon Sep 17 00:00:00 2001 From: TANG Ding Date: Mon, 24 Apr 2023 15:08:03 +0800 Subject: [PATCH] fix split error --- python/conformance/conformance_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/conformance/conformance_test.py b/python/conformance/conformance_test.py index 8a0d0b5..052205a 100644 --- a/python/conformance/conformance_test.py +++ b/python/conformance/conformance_test.py @@ -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 @@ -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: