From e4b590e826c759747f6125486b0e55a26aaaf16d Mon Sep 17 00:00:00 2001
From: Jacky <18255193+kthui@users.noreply.github.com>
Date: Thu, 16 May 2024 13:08:17 -0700
Subject: [PATCH] Fix L0_backend_python iGPU PyTorch installation (#7231)
 (#7232)

* Fix TEST_JETSON double square bracket issue

* Use and instead of or
---
 qa/L0_backend_python/examples/test.sh | 2 +-
 qa/L0_backend_python/test.sh          | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/qa/L0_backend_python/examples/test.sh b/qa/L0_backend_python/examples/test.sh
index ae8292cd0e..55ca569054 100755
--- a/qa/L0_backend_python/examples/test.sh
+++ b/qa/L0_backend_python/examples/test.sh
@@ -38,7 +38,7 @@ rm -fr *.log python_backend/
 
 # Install torch
 pip3 uninstall -y torch
-if [[ "$TEST_JETSON" == "0" ]] || [[ ${TEST_WINDOWS} == 0 ]]; then
+if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
     pip3 install torch==2.0.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0+cu117
 else
     pip3 install torch==2.0.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0
diff --git a/qa/L0_backend_python/test.sh b/qa/L0_backend_python/test.sh
index 8c5849bc61..98c88f7e9e 100755
--- a/qa/L0_backend_python/test.sh
+++ b/qa/L0_backend_python/test.sh
@@ -165,7 +165,7 @@ cp ../python_models/dlpack_identity/model.py ./models/dlpack_identity/1/
 cp ../python_models/dlpack_identity/config.pbtxt ./models/dlpack_identity
 
 
-if [[ "$TEST_JETSON" == "0" ]] && [[ ${TEST_WINDOWS} == 0 ]]; then
+if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
   pip3 install torch==1.13.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
 else
   # GPU tensor tests are disabled on jetson
@@ -237,7 +237,7 @@ set -e
 #
 # Test KIND_GPU
 # Disable env test for Jetson & Windows since GPU Tensors are not supported
-if [[ "$TEST_JETSON" == "0" ]] && [[ ${TEST_WINDOWS} == 0 ]]; then
+if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
   rm -rf models/
   mkdir -p models/add_sub_gpu/1/
   cp ../python_models/add_sub/model.py ./models/add_sub_gpu/1/
@@ -408,7 +408,7 @@ fi
 # Disable ensemble, io and bls tests for Jetson since GPU Tensors are not supported
 # Disable variants test for Jetson since already built without GPU Tensor support
 # Disable decoupled test because it uses GPU tensors
-if [[ "$TEST_JETSON" == "0" ]]; then
+if [ "$TEST_JETSON" == "0" ]; then
     SUBTESTS="ensemble bls decoupled"
     # [DLIS-6093] Disable variants test for Windows since tests are not executed in docker container (cannot apt update/install)
     # [DLIS-5970] Disable io tests for Windows since GPU Tensors are not supported