Skip to content

Commit

Permalink
Add rocm-dev
Browse files Browse the repository at this point in the history
  • Loading branch information
stellaraccident committed Aug 25, 2023
1 parent 7a8bb88 commit 4c7cefb
Showing 1 changed file with 20 additions and 0 deletions.
20 changes: 20 additions & 0 deletions dockerfiles/manylinux_x86_64.Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -8,3 +8,23 @@ RUN yum install -y epel-release && \
yum install -y capstone-devel tbb-devel libzstd-devel && \
yum clean all && \
rm -rf /var/cache/yum

######## AMD ROCM #######
ARG ROCM_VERSION=5.2.1
ARG AMDGPU_VERSION=22.20.1
ARG RHEL_VERSION=8.6

# Install the ROCm rpms
RUN echo -e "[ROCm]\nname=ROCm\nbaseurl=https://repo.radeon.com/rocm/yum/${ROCM_VERSION}/main\nenabled=1\ngpgcheck=0" >> /etc/yum.repos.d/rocm.repo \
&& echo -e "[amdgpu]\nname=amdgpu\nbaseurl=https://repo.radeon.com/amdgpu/${AMDGPU_VERSION}/rhel/${RHEL_VERSION}/main/x86_64\nenabled=1\ngpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo \
&& yum install -y rocm-dev \
&& yum clean all

######## GIT CONFIGURATION ########
# Git started enforcing strict user checking, which thwarts version
# configuration scripts in a docker image where the tree was checked
# out by the host and mapped in. Disable the check.
# See: https://github.com/openxla/iree/issues/12046
# We use the wildcard option to disable the checks. This was added
# in git 2.35.3
RUN git config --global --add safe.directory '*'

0 comments on commit 4c7cefb

Please sign in to comment.