Skip to content

Commit 6b32be0

Browse files
authored
ci: Add support for torch2.8 (#781)
Signed-off-by: oliver könig <okoenig@nvidia.com>
1 parent d49d7c9 commit 6b32be0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/_build.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -99,8 +99,8 @@ jobs:
9999
# e.g. we can have system CUDA version being 11.7 but if torch==1.12 then we need to download the wheel from cu116
100100
# This code is ugly, maybe there's a better way to do this.
101101
export TORCH_CUDA_VERSION=$(python -c "from os import environ as env; \
102-
minv = {'2.4': 118, '2.5': 118, '2.6': 118, '2.7': 118}[env['MATRIX_TORCH_VERSION']]; \
103-
maxv = {'2.4': 124, '2.5': 124, '2.6': 126, '2.7': 128}[env['MATRIX_TORCH_VERSION']]; \
102+
minv = {'2.4': 118, '2.5': 118, '2.6': 118, '2.7': 118, '2.8': 126}[env['MATRIX_TORCH_VERSION']]; \
103+
maxv = {'2.4': 124, '2.5': 124, '2.6': 126, '2.7': 128, '2.8': 129}[env['MATRIX_TORCH_VERSION']]; \
104104
print(minv if int(env['MATRIX_CUDA_VERSION']) < 120 else maxv)" \
105105
)
106106
if [[ ${{ inputs.torch-version }} == *"dev"* ]]; then

0 commit comments

Comments
 (0)