Skip to content
Snippets Groups Projects
Commit bba14765 authored by Yifan Zhao's avatar Yifan Zhao
Browse files

Merge remote-tracking branch 'origin/approx_hpvm_hotfix' into hpvm-release-exp

parents 9c3bbf76 baf4e498
No related branches found
No related tags found
No related merge requests found
......@@ -87,6 +87,9 @@ def parse_args(args=None):
parser.add_argument(
"-r", "--run-tests", action="store_true", help="Build and run test cases"
)
parser.add_argument(
"--no-pypkg", action="store_true", help="Don't build the HPVM Python Packages"
)
parser.add_argument(
"--no-params", action="store_true", help="Don't download DNN model parameters"
)
......@@ -158,6 +161,9 @@ Arguments: """
if args.cmake_args.strip() != "":
args.cmake_args = [f"-{arg}" for arg in args.cmake_args.split(" ")]
args.no_pypkg = not input_with_check(
"Install HPVM Python Packages (recommended)? [y/n]: ", parse_yn, "Please enter y or n"
)
args.no_params = not input_with_check(
"Download DNN weights (recommended)? [y/n]: ", parse_yn, "Please enter y or n"
)
......@@ -337,7 +343,8 @@ def main():
link_and_patch()
if not args.no_params:
check_download_model_params()
install_py_packages()
if not args.no_pypkg:
install_py_packages()
if args.no_build:
print(
"""
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment