from utils.log_update_handler import DBLogUpdateHandler
from utils.log_utils import trimLog, collectLogData, valid_interval
from utils.log_utils import DEFAULT_INTERVAL as default_interval
parser = argparse.ArgumentParser(description="Run the benchmark remotely")
parser.add_argument(
"--android_dir",
default="/data/local/tmp/",
help="The directory in the android device all files are pushed to.")
parser.add_argument(
"--app_id",
help="The app id you use to upload/download your file for everstore "
"and access the job queue")
parser.add_argument("--claimer_id",
default=getMachineId(),
type=claimer_id_type,
help="A unique claimer id to represent itself. "
"Must talk to Caffe2 team to set it up.")
parser.add_argument("--cooldown",
default=0,
type=float,
help="Specify the time interval between two test runs.")
parser.add_argument(
"-d",
"--devices",
help="Specify the devices to run the benchmark, in a comma separated "
"list. The value is the device or device_hash field of the meta info.")
parser.add_argument(
"--job_queue",
default="aibench_interactive",
from reboot_device import reboot as reboot_device
from utils.check_argparse import claimer_id_type
from utils.custom_logger import getLogger, setLoggerLevel
from utils.utilities import getFilename, getMachineId, setRunKilled
from utils.utilities import killed_flag as RUN_KILLED
from utils.watchdog import WatchDog
parser = argparse.ArgumentParser(description="Run the benchmark remotely")
parser.add_argument("--android_dir", default="/data/local/tmp/",
help="The directory in the android device all files are pushed to.")
parser.add_argument("--app_id",
help="The app id you use to upload/download your file for everstore "
"and access the job queue")
parser.add_argument("--claimer_id", default=getMachineId(),
type=claimer_id_type, help="A unique claimer id to represent itself. "
"Must talk to Caffe2 team to set it up.")
parser.add_argument("--cooldown", default=0, type=float,
help="Specify the time interval between two test runs.")
parser.add_argument("-d", "--devices",
help="Specify the devices to run the benchmark, in a comma separated "
"list. The value is the device or device_hash field of the meta info.")
parser.add_argument("--job_queue",
default="aibench_interactive",
help="Specify the db job queue that the benchmark is sent to")
parser.add_argument("--logger_level", default="info",
choices=["info", "warning", "error"],
help="Specify the logger level")
parser.add_argument("--model_cache", required=True,
help="The local directory containing the cached models. It should not "