And update trition version here as well

malfet · malfet · commit 8bcc10626ff1 · 2023-03-10T07:33:40.000-08:00
diff --git a/manywheel/build_cuda.sh b/manywheel/build_cuda.sh
@@ -234,7 +234,7 @@ fi
 # TODO: Remove me when Triton has a proper release channel
 if [[ $(uname) == "Linux" && -z "$PYTORCH_EXTRA_INSTALL_REQUIREMENTS" ]]; then
     TRITON_SHORTHASH=$(cut -c1-10 $PYTORCH_ROOT/.github/ci_commit_pins/triton.txt)
-    export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="pytorch-triton==2.0.0+${TRITON_SHORTHASH}"
+    export PYTORCH_EXTRA_INSTALL_REQUIREMENTS="pytorch-triton==2.1.0+${TRITON_SHORTHASH}"
 fi
 
 # builder/test.sh requires DESIRED_CUDA to know what tests to exclude
