[test-triton.sh] Allow USE_IPEX=0 when running benchmarks (#2765)

Signed-off-by: Whitney Tsang <[email protected]>
intel · Nov 20, 2024 · 5d1f804 · 5d1f804
1 parent 7551a90
commit 5d1f804
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/scripts/compile-pytorch-ipex.sh b/scripts/compile-pytorch-ipex.sh
@@ -66,7 +66,7 @@ if pip show torch &>/dev/null; then
     echo "**** Pinned PyTorch commit $PYTORCH_PINNED_COMMIT ****"
     if [ "$FORCE_REINSTALL" = false ]; then
       echo "**** Exiting without action. ****"
-      echo "**** INFO: Run the install-pytorch script with the --force-reinstall flag to force reinstallation of PyTorch,
+      echo "**** INFO: Run the compile-pytorch-ipex script with the --force-reinstall flag to force reinstallation of PyTorch,
         or uninstall the current version of PyTorch manually. ****"
       exit 1
     fi

diff --git a/scripts/test-triton.sh b/scripts/test-triton.sh
@@ -152,7 +152,7 @@ install_deps() {
     echo "**** Skipping installation of pytorch ****"
   else
     echo "**** Installing pytorch ****"
-    if [ "$TEST_BENCHMARK_SOFTMAX" = true ] || [ "$TEST_BENCHMARK_GEMM" = true ] || [ "$TEST_BENCHMARK_ATTENTION" = true ]; then
+    if ([ ! -v USE_IPEX ] || [ "$USE_IPEX" = 1 ]) && ([ "$TEST_BENCHMARK_SOFTMAX" = true ] || [ "$TEST_BENCHMARK_GEMM" = true ] || [ "$TEST_BENCHMARK_ATTENTION" = true ]); then
       $SCRIPTS_DIR/compile-pytorch-ipex.sh $([ $VENV = true ] && echo "--venv")
     else
       $SCRIPTS_DIR/install-pytorch.sh $([ $VENV = true ] && echo "--venv")