in benchmarking/run_lab.py [0:0]
def _getRawArgs(self):
"""Create raw args for BenchmarkDriver"""
try:
if "info" in self.job["benchmarks"]:
info = self.job["benchmarks"]["info"]
# pass the device hash as well as type
device = {"kind": self.job["device"], "hash": self.job["hash"]}
device_str = json.dumps(device)
raw_args = []
raw_args.extend(
[
"--benchmark_file",
self.job["benchmarks"]["benchmark"]["content"],
"--cooldown",
str(self.args.cooldown),
"--device",
device_str,
"--framework",
self.job["framework"],
"--info",
json.dumps(info),
"--model_cache",
self.args.model_cache,
"--platform",
self.args.platform,
"--remote_access_token",
self.args.remote_access_token,
"--root_model_dir",
self.args.root_model_dir,
"--simple_local_reporter",
self.tempdir,
"--user_identifier",
str(self.job["identifier"]),
"--user_string",
self.job.get("user"),
]
)
if self.job["framework"] != "generic":
raw_args.extend(["--remote_reporter", self.args.remote_reporter])
if self.args.shared_libs:
raw_args.extend(["--shared_libs", "'" + self.args.shared_libs + "'"])
if self.args.timeout:
raw_args.extend(["--timeout", str(self.args.timeout)])
if self.args.platform_sig:
raw_args.append("--platform_sig")
raw_args.append(self.args.platform_sig)
if self.args.monsoon_map:
raw_args.extend(["--monsoon_map", str(self.args.monsoon_map)])
if self.args.hash_platform_mapping:
# if the user provides filename, we will load it.
raw_args.append("--hash_platform_mapping")
raw_args.append(self.args.hash_platform_mapping)
if self.args.device_name_mapping:
raw_args.append("--device_name_mapping")
raw_args.append(self.args.device_name_mapping)
return raw_args
except Exception:
raise BenchmarkArgParseException("Error parsing raw args from job.")