From baf4e498419e8a93c50d4aced73686c7d4533829 Mon Sep 17 00:00:00 2001
From: hashimsharif <hashim.sharif91@gmail.com>
Date: Fri, 2 Apr 2021 23:48:47 -0500
Subject: [PATCH] Adding option to skip Python Package installation
---
hpvm/scripts/hpvm_installer.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/hpvm/scripts/hpvm_installer.py b/hpvm/scripts/hpvm_installer.py
index cce8b3f079..c9d9452330 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(
"""
--
GitLab