Skip to content

Commit 4ae9dbb

Browse files
committed
fix(torch): Include PyTorch patch for compatibility with CUDA 13.1.0
See: - pytorch/pytorch#169756 - pytorch/pytorch#164570
1 parent 5d2d31c commit 4ae9dbb

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

torch/Dockerfile

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,8 @@ RUN ./clone.sh pytorch/pytorch pytorch "${BUILD_TORCH_VERSION}" && \
5555
| git -C pytorch apply; \
5656
elif [ "${BUILD_TORCH_VERSION}" = '2.7.1' ]; then \
5757
git -C pytorch cherry-pick -n b74be524547f7f025066f19eda3b53a887c244ba; \
58+
elif [ "${BUILD_TORCH_VERSION}" = '2.9.1' ]; then \
59+
git -C pytorch cherry-pick -n a20afb61007a94f5c28294e9ae20043657152ef6; \
5860
fi && \
5961
rm -rf pytorch/.git
6062

0 commit comments

Comments
 (0)