pin the pytorch version to 2.8.0

Author: TroyGarden

.github/scripts/filter.py

@@ -62,6 +62,8 @@ def main():
     for entry in full_matrix["include"]:
         if entry["desired_cuda"] == "cu129":
             continue
+        # pin the pytorch version to 2.8.0
+        entry["stable_version"] = "2.8.0"
         new_matrix_entries.append(entry)
 
     new_matrix = {"include": new_matrix_entries}

.github/scripts/validate_binaries.sh

@@ -73,7 +73,7 @@ conda env config vars set -n ${CONDA_ENV}  \
 #     export PYTORCH_CUDA_PKG="pytorch-cuda=${MATRIX_GPU_ARCH_VERSION}"
 # fi
 
-conda run -n "${CONDA_ENV}" pip install torch --index-url "$PYTORCH_URL"
+conda run -n "${CONDA_ENV}" pip install torch==2.8.0 --index-url "$PYTORCH_URL"
 
 # install fbgemm
 conda run -n "${CONDA_ENV}" pip install fbgemm-gpu --index-url "$PYTORCH_URL"

.github/workflows/unittest_ci.yml

@@ -93,7 +93,7 @@ jobs:
           index_url=https://download.pytorch.org/whl/${{ inputs.channel }}/${{ matrix.cuda-tag }}
         fi
         conda run -n build_binary \
-          pip install torch --index-url $index_url
+          pip install torch==2.8.0 --index-url $index_url
         conda run -n build_binary \
           python -c "import torch; print(torch.__version__)"
         echo "torch succeeded"

.github/workflows/unittest_ci_cpu.yml

@@ -82,7 +82,7 @@ jobs:
           index_url=https://download.pytorch.org/whl/${{ inputs.channel }}/cpu
         fi
         conda run -n build_binary \
-          pip install torch --index-url $index_url
+          pip install torch==2.8.0 --index-url $index_url
         conda run -n build_binary \
           python -c "import torch"
         echo "torch succeeded"