Skip to content

Commit

Permalink
Merge pull request #2558 from ROCm/bump_rocm611
Browse files Browse the repository at this point in the history
Bump TF to ROCm 6.1.1
  • Loading branch information
i-chaochen authored Jun 4, 2024
2 parents 8d06bdb + b1a42c9 commit 82c8caa
Show file tree
Hide file tree
Showing 8 changed files with 9 additions and 9 deletions.
2 changes: 1 addition & 1 deletion build_rocm_python3
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ shift "$((OPTIND-1))"
nightly=true

# First positional argument (if any) specifies the ROCM_INSTALL_DIR
ROCM_INSTALL_DIR=/opt/rocm-6.1.0
ROCM_INSTALL_DIR=/opt/rocm-6.1.1
if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
fi
Expand Down
2 changes: 1 addition & 1 deletion rocm_docs/tensorflow-build-from-source.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,7 @@ For details of the ROCm instructions, please refer to the [ROCm QuickStart Insta

To build with ROCm3.10, set the following environment variables, and add those environment variables at the end of ~/.bashrc
```
export ROCM_PATH=/opt/rocm-6.1.0
export ROCM_PATH=/opt/rocm-6.1.1
export HCC_HOME=$ROCM_PATH/hcc
export HIP_PATH=$ROCM_PATH/hip
export PATH=$HCC_HOME/bin:$HIP_PATH/bin:$PATH
Expand Down
4 changes: 2 additions & 2 deletions tensorflow/tools/ci_build/Dockerfile.rocm
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,10 @@
FROM ubuntu:focal
MAINTAINER Jeff Poznanovic <jeffrey.poznanovic@amd.com>

ARG ROCM_DEB_REPO=https://repo.radeon.com/rocm/apt/6.1/
ARG ROCM_DEB_REPO=https://repo.radeon.com/rocm/apt/6.1.1/
ARG ROCM_BUILD_NAME=ubuntu
ARG ROCM_BUILD_NUM=main
ARG ROCM_PATH=/opt/rocm-6.1.0
ARG ROCM_PATH=/opt/rocm-6.1.1
ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100"

ARG DEBIAN_FRONTEND=noninteractive
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/linux/rocm/run_gpu_multi.sh
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
else
if [[ -z "${ROCM_PATH}" ]]; then
ROCM_INSTALL_DIR=/opt/rocm-6.1.0
ROCM_INSTALL_DIR=/opt/rocm-6.1.1
else
ROCM_INSTALL_DIR=$ROCM_PATH
fi
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/linux/rocm/run_gpu_single.sh
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
else
if [[ -z "${ROCM_PATH}" ]]; then
ROCM_INSTALL_DIR=/opt/rocm-6.1.0
ROCM_INSTALL_DIR=/opt/rocm-6.1.1
else
ROCM_INSTALL_DIR=$ROCM_PATH
fi
Expand Down
2 changes: 1 addition & 1 deletion tensorflow/tools/ci_build/linux/rocm/run_xla.sh
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ if [[ -n $1 ]]; then
ROCM_INSTALL_DIR=$1
else
if [[ -z "${ROCM_PATH}" ]]; then
ROCM_INSTALL_DIR=/opt/rocm-6.1.0
ROCM_INSTALL_DIR=/opt/rocm-6.1.1
else
ROCM_INSTALL_DIR=$ROCM_PATH
fi
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,7 @@ RUN /setup.packages.sh /runtime.packages.txt
# Install ROCM
ARG TF_PKGS_DIR=tmp/packages
ARG TENSORFLOW_PACKAGE=tf_nightly_rocm
ARG ROCM_VERSION=6.1.0
ARG ROCM_VERSION=6.1.1
ARG CUSTOM_INSTALL
ARG ROCM_PATH=/opt/rocm-${ROCM_VERSION}
ENV ROCM_PATH=${ROCM_PATH}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ COPY builder.packages.txt builder.packages.txt
RUN /setup.packages.sh /builder.packages.txt

# Install ROCM
ARG ROCM_VERSION=6.0.2
ARG ROCM_VERSION=6.1.1
ARG CUSTOM_INSTALL
ARG ROCM_PATH=/opt/rocm-${ROCM_VERSION}
ENV ROCM_PATH=${ROCM_PATH}
Expand Down

0 comments on commit 82c8caa

Please sign in to comment.