From d2e7b24ded4803b64e42a0926a829ccf1af8a2ec Mon Sep 17 00:00:00 2001
From: atalman <atalman@fb.com>
Date: Mon, 25 Mar 2024 11:39:12 -0700
Subject: [PATCH] [Release Only] Use pytorch-test channel when testing conda
 builds

---
 conda/build_pytorch.sh | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/conda/build_pytorch.sh b/conda/build_pytorch.sh
index 1cc09db85..ed999fd32 100755
--- a/conda/build_pytorch.sh
+++ b/conda/build_pytorch.sh
@@ -408,7 +408,13 @@ for py_ver in "${DESIRED_PYTHON[@]}"; do
         else
           local_channel="$(pwd)/$output_folder"
         fi
-        conda install -y -c "file://$local_channel" pytorch==$PYTORCH_BUILD_VERSION -c pytorch -c numba/label/dev -c pytorch-nightly -c nvidia
+
+        CONDA_CHANNEL="pytorch-test"
+        if [[ -n "$OVERRIDE_PACKAGE_VERSION" && "$OVERRIDE_PACKAGE_VERSION" =~ .*dev.* ]]; then
+            CONDA_CHANNEL="pytorch-nightly"
+        fi
+
+        conda install -y -c "file://$local_channel" pytorch==$PYTORCH_BUILD_VERSION -c pytorch -c numba/label/dev -c $CONDA_CHANNEL -c nvidia
 
         echo "$(date) :: Running tests"
         pushd "$pytorch_rootdir"