diff --git a/tests/st/broadcast/test_broadcast_auto_parallel.py b/tests/st/broadcast/test_broadcast_auto_parallel.py index 576cde6d4c..1b5ca8caef 100644 --- a/tests/st/broadcast/test_broadcast_auto_parallel.py +++ b/tests/st/broadcast/test_broadcast_auto_parallel.py @@ -24,4 +24,5 @@ import pytest def test_broadcast_auto_parallel(): sh_path = os.path.split(os.path.realpath(__file__))[0] ret = os.system(f"sh {sh_path}/run_broadcast_auto_parallel.sh") + os.system(f"grep -E 'ERROR|error' {sh_path}/lenet_broadcast*/test_lenet_auto_parallel_broadcast_8p_log*log -C 3") assert ret == 0 diff --git a/tests/st/mix_precision/test_mix_precision.py b/tests/st/mix_precision/test_mix_precision.py index 5ba8f83725..5796efb1b2 100644 --- a/tests/st/mix_precision/test_mix_precision.py +++ b/tests/st/mix_precision/test_mix_precision.py @@ -36,7 +36,6 @@ def read_validateir_file(path_folder): filename = find_newest_validateir_file(path_folder) with open(os.path.join(filename), 'r') as f: contend = f.read() - clean_all_ir_files(path_folder) return contend @@ -130,10 +129,12 @@ def test_sit_auto_mix_precision_model_o0(): contend = read_validateir_file('./test_amp_o0') castnum = re.findall("Cast", contend) assert len(castnum) == 5 + clean_all_ir_files('./test_amp_o0') model.predict(Tensor(input_data)) contend = read_validateir_file('./test_amp_o0') castnum = re.findall("Cast", contend) assert len(castnum) == 11 + clean_all_ir_files('./test_amp_o0') @pytest.mark.level0 @@ -164,6 +165,7 @@ def test_sit_auto_mix_precision_model_o2(): contend = read_validateir_file('./test_amp_o2') castnum = re.findall("Cast", contend) assert len(castnum) == 14 + clean_all_ir_files('./test_amp_o2') out_graph = model.predict(Tensor(input_data)) # pynative mode