Skip to content

Commit a1c35e7

Browse files
authored
Using torch commit that supports running scaled_mm on Radeon (#492)
1 parent 7210905 commit a1c35e7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Dockerfile.rocm_base

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ ARG RCCL_BRANCH="648a58d"
66
ARG RCCL_REPO="https://github.com/ROCm/rccl"
77
ARG TRITON_BRANCH="e5be006"
88
ARG TRITON_REPO="https://github.com/triton-lang/triton.git"
9-
ARG PYTORCH_BRANCH="6c0e7463"
9+
ARG PYTORCH_BRANCH="295f2ed4"
1010
ARG PYTORCH_VISION_BRANCH="v0.21.0"
1111
ARG PYTORCH_REPO="https://github.com/pytorch/pytorch.git"
1212
ARG PYTORCH_VISION_REPO="https://github.com/pytorch/vision.git"

0 commit comments

Comments
 (0)