diff --git a/hpvm/scripts/hpvm_installer.py b/hpvm/scripts/hpvm_installer.py index cce8b3f07928d9ab096df3166bd02f4e6f8e1f5d..c9d94523301e1847c7fd8227700d7542a3db1e06 100755 --- a/hpvm/scripts/hpvm_installer.py +++ b/hpvm/scripts/hpvm_installer.py @@ -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( """