Skip to content

Commit df23042

Browse files
committed
add binary validation for python 3.13t (freethreading)
1 parent 89d6ae0 commit df23042

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

.github/scripts/validate_binaries.sh

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -10,22 +10,18 @@ export PYTORCH_CUDA_PKG=""
1010
export CONDA_ENV="build_binary"
1111

1212
if [[ ${MATRIX_PYTHON_VERSION} = '3.13t' ]]; then
13-
echo "Conda doesn't support 3.13t yet, you can just try \`conda create -n test python=3.13t\`"
14-
exit 0
13+
# use conda-forge to install python3.13t
14+
conda create -y -n "${CONDA_ENV}" python="3.13" python-freethreading -c conda-forge
15+
else
16+
conda create -y -n "${CONDA_ENV}" python="${MATRIX_PYTHON_VERSION}"
1517
fi
1618

17-
conda create -y -n "${CONDA_ENV}" python="${MATRIX_PYTHON_VERSION}"
18-
1919
conda run -n build_binary python --version
2020

2121
# Install pytorch, torchrec and fbgemm as per
2222
# installation instructions on following page
2323
# https://github.com/pytorch/torchrec#installations
2424

25-
if [[ ${MATRIX_GPU_ARCH_TYPE} = 'rocm' ]]; then
26-
echo "We don't support rocm"
27-
exit 0
28-
fi
2925

3026
# figure out CUDA VERSION
3127
if [[ ${MATRIX_GPU_ARCH_TYPE} = 'cuda' ]]; then

0 commit comments

Comments
 (0)