add max_timeout for cmd

AI-performance · Aug 9, 2020 · c968a60 · c968a60
1 parent e61debb
commit c968a60
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/core/global_config.py b/core/global_config.py
@@ -6,14 +6,17 @@
 sys.path.append("..")
 from utils.log import LoggerCreator  # noqa
 
+##############################
+# Global Config
+##############################
 log_enable_debug = True
 logger_creator = LoggerCreator(log_enable_debug)
 logger = logger_creator.create_logger()
 GPU_REPEATS = 1000  # 1000
 CPU_REPEATS = 100  # 100
 WARMUP = 20  # 20
-MAX_TIMEOUT_SECOND = 250
-MAX_TIMEOUT_SECOND_ONCE_INFER = 0.5
+MAX_TIMEOUT_SECOND = 10  # 10, not used for infer command
+MAX_TIMEOUT_SECOND_ONCE_INFER = 0.5  # used to calc MAX_TIMEOUT_SECOND
 
 
 def create_config(framework_name):

diff --git a/utils/cmd.py b/utils/cmd.py
@@ -8,10 +8,10 @@
 import unittest
 
 sys.path.append("..")
-from core.global_config import logger  # noqa
+from core.global_config import logger, MAX_TIMEOUT_SECOND  # noqa
 
 
-def run_cmd(cmd, wait_interval_sec=5, max_timeout_sec=100):
+def run_cmd(cmd, wait_interval_sec=5, max_timeout_sec=MAX_TIMEOUT_SECOND):
     cmd_type = "CMD"
     logger.info("{}> {}".format(cmd_type, cmd))
     subp = subprocess.Popen(