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): ...@@ -87,6 +87,9 @@ def parse_args(args=None):
parser.add_argument( parser.add_argument(
"-r", "--run-tests", action="store_true", help="Build and run test cases" "-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( parser.add_argument(
"--no-params", action="store_true", help="Don't download DNN model parameters" "--no-params", action="store_true", help="Don't download DNN model parameters"
) )
...@@ -158,6 +161,9 @@ Arguments: """ ...@@ -158,6 +161,9 @@ Arguments: """
if args.cmake_args.strip() != "": if args.cmake_args.strip() != "":
args.cmake_args = [f"-{arg}" for arg in args.cmake_args.split(" ")] 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( args.no_params = not input_with_check(
"Download DNN weights (recommended)? [y/n]: ", parse_yn, "Please enter y or n" "Download DNN weights (recommended)? [y/n]: ", parse_yn, "Please enter y or n"
) )
...@@ -337,7 +343,8 @@ def main(): ...@@ -337,7 +343,8 @@ def main():
link_and_patch() link_and_patch()
if not args.no_params: if not args.no_params:
check_download_model_params() check_download_model_params()
install_py_packages() if not args.no_pypkg:
install_py_packages()
if args.no_build: if args.no_build:
print( 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