Using pytorch commit past the point when rowwise PR (pytorch/pytorch#…

…144432) was merged (ROCm#384)
EmbeddedLLM · Jan 27, 2025 · 6612f91 · 6612f91
1 parent 718ac48
commit 6612f91
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile.base b/Dockerfile.base
@@ -5,7 +5,7 @@ ARG RCCL_BRANCH="648a58d"
 ARG RCCL_REPO="https://github.com/ROCm/rccl"
 ARG TRITON_BRANCH="e5be006"
 ARG TRITON_REPO="https://github.com/triton-lang/triton.git"
-ARG PYTORCH_BRANCH="8d4926e"
+ARG PYTORCH_BRANCH="3a585126"
 ARG PYTORCH_VISION_BRANCH="v0.19.1"
 ARG PYTORCH_REPO="https://github.com/pytorch/pytorch.git"
 ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"