def openvino_root_dir(request):
"""Fixture function for command-line option."""
openvino_root_dir = request.config.getoption("openvino_root_dir",
skip=True)
validate_path_arg(openvino_root_dir, is_dir=True)
return openvino_root_dir
def openvino_ref(request, artifacts):
"""Fixture function for command-line option.
Return path to root directory with installed OpenVINO.
If --openvino_ref command-line option is not specified firstly build and install
instrumented package with OpenVINO repository specified in --openvino_root_dir option.
"""
openvino_ref = request.config.getoption("openvino_ref")
if openvino_ref:
validate_path_arg(openvino_ref, is_dir=True)
return openvino_ref
openvino_root_dir = request.config.getoption("openvino_root_dir", skip=True)
validate_path_arg(openvino_root_dir, is_dir=True)
build_dir = openvino_root_dir / "build_instrumented"
openvino_ref_path = artifacts / "ref_pkg"
log.info("--openvino_ref is not specified. Preparing instrumented build at %s", build_dir)
build_target = {"sea_itt_lib": Path(build_dir / "thirdparty" / "itt_collector" / "sea_itt_lib")}
return_code, output = make_build(
openvino_root_dir,
build_dir,
openvino_ref_path,
build_target=build_target,
cmake_additional_args=["-DSELECTIVE_BUILD=COLLECT"],
log=log
)
assert return_code == 0, f"Command exited with non-zero status {return_code}:\n {output}"
return openvino_ref_path
def omz_cache_dir(request):
"""Fixture function for command-line option."""
omz_cache_dir = request.config.getoption("omz_cache_dir", skip=True)
if omz_cache_dir:
try:
validate_path_arg(omz_cache_dir, is_dir=True)
except ValueError:
log.warning(f'The Open Model Zoo cache directory'
f' "{omz_cache_dir}" does not exist.')
return omz_cache_dir
def omz_repo(request):
"""Fixture function for command-line option."""
omz_repo = request.config.getoption("omz_repo", skip=True)
validate_path_arg(omz_repo, is_dir=True)
return omz_repo
def collector_dir(request):
"""Fixture function for command-line option."""
collector_dir = request.config.getoption("collector_dir", skip=True)
validate_path_arg(collector_dir, is_dir=True)
return collector_dir
def sea_runtool(request):
"""Fixture function for command-line option."""
sea_runtool = request.config.getoption("sea_runtool", skip=True)
validate_path_arg(sea_runtool)
return sea_runtool
