Skip to content

Commit 4ebb8be

Browse files
authored
export minv and maxv for pytorch 2.6
1 parent 7104b34 commit 4ebb8be

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/publish.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,8 +118,8 @@ jobs:
118118
# see https://github.com/pytorch/pytorch/blob/main/RELEASE.md#release-compatibility-matrix
119119
# This code is ugly, maybe there's a better way to do this.
120120
export TORCH_CUDA_VERSION=$(python -c "from os import environ as env; \
121-
minv = {'2.0': 117, '2.1': 118, '2.2': 118, '2.3': 118, '2.4': 118, '2.5': 118}[env['MATRIX_TORCH_VERSION']]; \
122-
maxv = {'2.0': 118, '2.1': 121, '2.2': 121, '2.3': 121, '2.4': 121, '2.5': 124}[env['MATRIX_TORCH_VERSION']]; \
121+
minv = {'2.0': 117, '2.1': 118, '2.2': 118, '2.3': 118, '2.4': 118, '2.5': 118, '2.6': 118}[env['MATRIX_TORCH_VERSION']]; \
122+
maxv = {'2.0': 118, '2.1': 121, '2.2': 121, '2.3': 121, '2.4': 121, '2.5': 124, '2.6': 126}[env['MATRIX_TORCH_VERSION']]; \
123123
print(max(min(int(env['MATRIX_CUDA_VERSION']), maxv), minv))" \
124124
)
125125
if [[ ${{ matrix.torch-version }} == *"dev"* ]]; then

0 commit comments

Comments
 (0)