remove lib if needed

Signed-off-by: yiliu30 <yi4.liu@intel.com>
intel · Dec 27, 2024 · 5dbc191 · 5dbc191
1 parent 582121e
commit 5dbc191
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -253,11 +253,11 @@ def detect_local_sm_architectures():
     # 1. python setup.py lib install
     # 2. Within the gaudi docker where the HPU is available, we install the auto_round_lib by default.
     is_user_requesting_library_build = "lib" in sys.argv
-
+    if is_habana_framework_installed:
+        sys.argv.remove("lib")
     should_build_library = is_user_requesting_library_build or BUILD_HPU_ONLY
 
     if should_build_library:
-        sys.argv.remove("lib")
         package_name = "auto_round_lib"
         INSTALL_CFG = LIB_INSTALL_CFG
     else: