"src/include/ConstantTensorDescriptor.cuh" did not exist on "ac1f62be3fbdad4f37d98ad0928914fe591a6364"
Unverified Commit 889a6b26 authored by Hubert's avatar Hubert Committed by GitHub
Browse files

[Fix] fix log re-direct (#564)

parent cf5a6d1a
from mmengine.config import read_base
from opencompass.models.minimax import MiniMax
from opencompass.models import MiniMax
from opencompass.partitioners import NaivePartitioner
from opencompass.runners import LocalRunner
from opencompass.runners.local_api import LocalAPIRunner
......
......@@ -4,6 +4,7 @@ import os.path as osp
import subprocess
import sys
import time
import traceback
from multiprocessing import Manager, Pool
from multiprocessing.managers import SyncManager
from typing import Any, Dict, List, Tuple
......@@ -122,9 +123,13 @@ def launch(task: BaseTask, tokens: SyncManager.Semaphore):
end_time = time.time()
logger.info(f'time elapsed: {end_time - start_time:.2f}s')
except Exception:
# print trace back in target file
traceback.print_exc()
# reset stdout and stderr
reset_std()
logger.warning(f'task {task_name} fail, see\n{out_path}')
returncode = 1
finally:
else:
# reset stdout and stderr
reset_std()
return task_name, returncode
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment