Skip to content

Commit

Permalink
navi lib versions
Browse files Browse the repository at this point in the history
  • Loading branch information
gshtras committed Dec 20, 2024
1 parent 70142eb commit c04381a
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions Dockerfile.base_navi
Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
ARG BASE_IMAGE=rocm/dev-ubuntu-22.04:6.3-complete
ARG HIPBLASLT_BRANCH="d43d84a"
ARG HIPBLASLT_BRANCH="4d40e36"
ARG LEGACY_HIPBLASLT_OPTION=
ARG RCCL_BRANCH="dfe4a3e"
ARG RCCL_BRANCH="648a58d"
ARG RCCL_REPO="https://github.com/ROCm/rccl"
ARG TRITON_BRANCH="e192dba"
ARG TRITON_BRANCH="e5be006"
ARG TRITON_REPO="https://github.com/triton-lang/triton.git"
ARG PYTORCH_BRANCH="8bc4033"
ARG PYTORCH_BRANCH="8d4926e"
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"
Expand Down

0 comments on commit c04381a

Please sign in to comment.