reverting the script change

intel · Jan 12, 2024 · b9a782d · b9a782d
1 parent 0c6269f
commit b9a782d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/scripts/test-triton.sh b/scripts/test-triton.sh
@@ -53,8 +53,7 @@ export TRITON_PROJ_BUILD=$TRITON_PROJ/python/build
 
 python3 -m pip install lit
 python3 -m pip install pytest
-#python3 -m pip install torch==1.13.0a0+git6c9b55e intel_extension_for_pytorch==1.13.120+xpu -f https://developer.intel.com/ipex-whl-stable-xpu
-python3 -m pip install torch==2.0.1a0 torchvision==0.15.2a0 intel_extension_for_pytorch==2.0.110+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
+python3 -m pip install torch==1.13.0a0+git6c9b55e intel_extension_for_pytorch==1.13.120+xpu -f https://developer.intel.com/ipex-whl-stable-xpu
 if [ $? -ne 0 ]; then
   echo "FAILED: return code $?"
   exit $?