compile-pytorch-ipex.sh should install upstream pytorch by default (#…

…2022) Closes #2020 --------- Signed-off-by: Kirill Suvorov <[email protected]>
intel · Aug 27, 2024 · 386ae76 · 386ae76
1 parent fa98f75
commit 386ae76
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/scripts/compile-pytorch-ipex.sh b/scripts/compile-pytorch-ipex.sh
@@ -83,6 +83,16 @@ if [ "$UPSTREAM_PYTORCH" = true ] && [ "$BUILD_IPEX" = true ]; then
   exit 1
 fi
 
+if [ "$BUILD_PYTORCH" = false ] && [ "$UPSTREAM_PYTORCH" = false ]; then
+  echo "***** Use upstream pytorch by the default *****"
+  UPSTREAM_PYTORCH=true
+fi
+
+if [ "$BUILD_PINNED" = false ] && [ "$BUILD_FROM_SOURCE" = false ]; then
+  echo "***** Use pinned pytorch by the default *****"
+  BUILD_PINNED=true
+fi
+
 if [ ! -v BASE ]; then
   echo "**** BASE is not given *****"
   BASE=$ROOT/.scripts_cache
@@ -189,11 +199,6 @@ if [ "$BUILD_FROM_SOURCE" = false ]; then
   fi
 fi
 
-if [ "$BUILD_PINNED" = false ]; then
-  echo "**** Since '--pinned' option is not used, enable building from source. ****"
-  BUILD_FROM_SOURCE=true
-fi
-
 if [ "$BUILD_FROM_SOURCE" = false ]; then
   echo "**** Download nightly builds. ****"
   PYTHON_VERSION=$( python -c "import sys; print(f'{sys.version_info.major}.{sys.version_info.minor}')" )