Skip to content
Snippets Groups Projects
Commit baf4e498 authored by hashimsharif's avatar hashimsharif
Browse files

Adding option to skip Python Package installation

parent f7bf857e
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