File tree Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -71,22 +71,24 @@ jobs:
7171 - windows-latest
7272 - macos-latest
7373 python-version :
74- - " 3.8"
7574 - " 3.9"
7675 - " 3.10"
7776 - " 3.11"
7877 pytorch-computation-backend :
7978 - cpu
80- - cu117
8179 - cu118
80+ - cu121
81+ - cu124
8282 exclude :
83- - os : macos-latest
84- pytorch-computation-backend : cu117
8583 - os : macos-latest
8684 pytorch-computation-backend : cu118
85+ - os : macos-latest
86+ pytorch-computation-backend : cu121
87+ - os : macos-latest
88+ pytorch-computation-backend : cu124
8789 # TODO: find a way to test this
8890 # - os: ubuntu-latest
89- # pytorch-computation-backend: rocm5.4 .2
91+ # pytorch-computation-backend: rocm6 .2
9092 fail-fast : false
9193
9294 runs-on : ${{ matrix.os }}
@@ -108,7 +110,7 @@ jobs:
108110 - name : Install torch
109111 run :
110112 ltt install --pytorch-computation-backend=${{
111- matrix.pytorch-computation-backend }} torch==2.0.0
113+ matrix.pytorch-computation-backend }} torch==2.5.1
112114
113115 - name : Check computation backend
114116 shell : python
You can’t perform that action at this time.
0 commit comments