Skip to content

Commit c165c1a

Browse files
committed
update compute platforms list
proposed initially in #12
1 parent 0b98daa commit c165c1a

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

update-pytorch.sh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -115,11 +115,11 @@ then
115115
fi
116116
echo "updates found in PyTorch main index: updating also compute platform specific ones..."
117117

118-
# update compute-platform specific indexes
118+
# update compute-platform specific indexes: see current "Compute Platforms" on https://pytorch.org/
119119
# ignore old non-updated ones:
120-
# - cu75 cu80 cu90 cu91 cu92 cu100 cu101 cu102 cu110 cu111 cu113 cu115 cu116 cu117 cu117_pypi_cudnn
121-
# - rocm3.10 rocm3.7 rocm3.8 rocm4.0.1 rocm4.1 rocm4.2 rocm4.3.1 rocm4.5.2 rocm5.0 rocm5.1.1 rocm5.2 rocm5.3 rocm5.4.2 rocm5.5 rocm5.6 rocm5.7
122-
for d in cpu cpu-cxx11-abi cpu_pypi_pkg cu118 cu121 cu124 rocm6.0 rocm6.1
120+
# - cu75 cu80 cu90 cu91 cu92 cu100 cu101 cu102 cu110 cu111 cu113 cu115 cu116 cu117 cu117_pypi_cudnn cu121 cu124
121+
# - rocm3.10 rocm3.7 rocm3.8 rocm4.0.1 rocm4.1 rocm4.2 rocm4.3.1 rocm4.5.2 rocm5.0 rocm5.1.1 rocm5.2 rocm5.3 rocm5.4.2 rocm5.5 rocm5.6 rocm5.7 rocm6.0 rocm6.1
122+
for d in cpu cpu-cxx11-abi cpu_pypi_pkg cu118 cu126 cu128 rocm6.3 xpu
123123
do
124124
updateIndex "whl/$d"
125125
done

0 commit comments

Comments
 (0)