diff --git a/tools/converter.py b/tools/converter.py index c1afcb58354a813164da8c36b214be19a501dff4..8bb9adad2d6f1a1db75c2a7c6fb4ee1470495d7b 100644 --- a/tools/converter.py +++ b/tools/converter.py @@ -14,12 +14,9 @@ import argparse import glob -import hashlib -import os -import re import sh import sys -import urllib +import time import yaml from enum import Enum @@ -1006,8 +1003,11 @@ def run_mace(flags): flags.address_sanitizer, flags.mace_lib_type) # run + start_time = time.time() with device.lock(): device.run_specify_abi(flags, configs, target_abi) + elapse_minutes = (time.time() - start_time) / 60 + print("Elapse time: %f minutes." % elapse_minutes) elif dev[YAMLKeyword.device_name] != SystemType.host: six.print_('The device with soc %s do not support abi %s' % (dev[YAMLKeyword.target_socs], target_abi), @@ -1090,8 +1090,11 @@ def benchmark_model(flags): not flags.disable_openmp, flags.mace_lib_type) device = DeviceWrapper(dev) + start_time = time.time() with device.lock(): device.bm_specific_target(flags, configs, target_abi) + elapse_minutes = (time.time() - start_time) / 60 + print("Elapse time: %f minutes." % elapse_minutes) else: six.print_('There is no abi %s with soc %s' % (target_abi, dev[YAMLKeyword.target_socs]), diff --git a/tools/sh_commands.py b/tools/sh_commands.py index 11d6934beed8e269c7bad27fad9f718fa2d260e1..0649129f76db79860250b558a4cd7745df853c9a 100644 --- a/tools/sh_commands.py +++ b/tools/sh_commands.py @@ -69,7 +69,7 @@ def device_lock_path(serialno): return "/tmp/device-lock-%s" % serialno -def device_lock(serialno, timeout=3600): +def device_lock(serialno, timeout=7200): import filelock return filelock.FileLock(device_lock_path(serialno), timeout=timeout)