Skip to content

Commit a701022

Browse files
JehandadKhanJehandadKhan
authored
add gfx1101 target (#249)
1 parent 15255dd commit a701022

File tree

7 files changed

+70
-71
lines changed

7 files changed

+70
-71
lines changed

‎build/build.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -241,7 +241,7 @@ def add_artifact_subcommand_arguments(parser: argparse.ArgumentParser):
241241
rocm_group.add_argument(
242242
"--rocm_amdgpu_targets",
243243
type=str,
244-
default="gfx900,gfx906,gfx908,gfx90a,gfx940,gfx941,gfx942,gfx1030,gfx1100,gfx1200,gfx1201",
244+
default="gfx900,gfx906,gfx908,gfx90a,gfx940,gfx941,gfx942,gfx1030,gfx1100,gfx1101,gfx1200,gfx1201",
245245
help="A comma-separated list of ROCm amdgpu targets to support.",
246246
)
247247

‎build/rocm/Dockerfile.ms‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ RUN apt-get update && apt-get install -y \
1313
&& rm -rf /var/lib/apt/lists/*
1414
1515
# Add target file to help determine which device(s) to build for
16-
ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1200 gfx1201"
16+
ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201"
1717
ENV GPU_DEVICE_TARGETS=${GPU_DEVICE_TARGETS}
1818
1919
# Install ROCm

‎build/rocm/build_wheels/Dockerfile.manylinux_2_28_x86_64.rocm‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ RUN --mount=type=cache,target=/var/cache/dnf \
1515
--mount=type=bind,source=build/rocm/tools/get_rocm.py,target=get_rocm.py \
1616
python3 get_rocm.py --rocm-version=$ROCM_VERSION --job-name=$ROCM_BUILD_JOB --build-num=$ROCM_BUILD_NUM
1717

18-
ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1200 gfx1201"
18+
ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201"
1919
RUN printf '%s\n' > /opt/rocm/bin/target.lst ${GPU_DEVICE_TARGETS}
2020

2121
# Install LLVM 18 and dependencies.

‎build/rocm/docker/Dockerfile.jax-ubu22‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ RUN apt-get update && apt-get install -y \
1010
&& rm -rf /var/lib/apt/lists/*
1111

1212
# Add target file to help determine which device(s) to build for
13-
ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1200 gfx1201"
13+
ARG GPU_DEVICE_TARGETS="gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201"
1414
ENV GPU_DEVICE_TARGETS=${GPU_DEVICE_TARGETS}
1515

1616
# Install ROCM

‎build/rocm/docker/Dockerfile.jax-ubu24‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ RUN apt-get update && apt-get install -y \
1010
&& rm -rf /var/lib/apt/lists/*
1111

1212
# Add target file to help determine which device(s) to build for
13-
ARG GPU_DEVICE_TARGETS="gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1200 gfx1201"
13+
ARG GPU_DEVICE_TARGETS="gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201"
1414
ENV GPU_DEVICE_TARGETS=${GPU_DEVICE_TARGETS}
1515

1616
# Install ROCM

‎build/rocm/setup.rocm.sh‎

Lines changed: 64 additions & 65 deletions
Original file line numberDiff line numberDiff line change
@@ -16,77 +16,76 @@ ROCM_BUILD_NUM=main
1616
# Intial release don't have the trialing '.0'
1717
# For example ROCM 5.7.0 is at https://repo.radeon.com/rocm/apt/5.7/
1818
if [ ${ROCM_VERSION##*[^0-9]} -eq '0' ]; then
19-
ROCM_VERS=${ROCM_VERSION%.*}
19+
ROCM_VERS=${ROCM_VERSION%.*}
2020
else
21-
ROCM_VERS=$ROCM_VERSION
21+
ROCM_VERS=$ROCM_VERSION
2222
fi
2323
ROCM_DEB_REPO=${ROCM_DEB_REPO_HOME}${ROCM_VERS}/
2424

2525
if [ ! -f "/${CUSTOM_INSTALL}" ]; then
26-
# Add rocm repository
27-
chmod 1777 /tmp
28-
DEBIAN_FRONTEND=noninteractive apt-get --allow-unauthenticated update
29-
DEBIAN_FRONTEND=noninteractive apt install -y wget software-properties-common
30-
DEBIAN_FRONTEND=noninteractive apt-get clean all
31-
wget -qO - https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -;
32-
if [[ $ROCM_DEB_REPO == https://repo.radeon.com/rocm/* ]] ; then \
33-
echo "deb [arch=amd64] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/rocm.list; \
34-
else \
35-
echo "deb [arch=amd64 trusted=yes] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" > /etc/apt/sources.list.d/rocm.list ; \
36-
fi
37-
#Install rocm and other packages
38-
apt-get update --allow-insecure-repositories && DEBIAN_FRONTEND=noninteractive apt-get install -y \
39-
build-essential \
40-
software-properties-common \
41-
clang-6.0 \
42-
clang-format-6.0 \
43-
curl \
44-
g++-multilib \
45-
git \
46-
vim \
47-
libnuma-dev \
48-
virtualenv \
49-
python3-pip \
50-
pciutils \
51-
python-is-python3 \
52-
libffi-dev \
53-
libssl-dev \
54-
build-essential \
55-
zlib1g-dev \
56-
libbz2-dev \
57-
libreadline-dev \
58-
libsqlite3-dev curl \
59-
libncursesw5-dev \
60-
xz-utils \
61-
tk-dev \
62-
libxml2-dev \
63-
libxmlsec1-dev \
64-
libffi-dev \
65-
liblzma-dev \
66-
wget \
67-
rocm-dev \
68-
rocm-libs \
69-
miopen-hip \
70-
miopen-hip-dev \
71-
rocblas \
72-
rocblas-dev \
73-
rocsolver-dev \
74-
rocrand-dev \
75-
rocfft-dev \
76-
hipfft-dev \
77-
hipblas-dev \
78-
rocprim-dev \
79-
hipcub-dev \
80-
rccl-dev \
81-
hipsparse-dev \
82-
hipsolver-dev \
83-
wget && \
84-
apt-get clean && \
85-
rm -rf /var/lib/apt/lists/*
86-
26+
# Add rocm repository
27+
chmod 1777 /tmp
28+
DEBIAN_FRONTEND=noninteractive apt-get --allow-unauthenticated update
29+
DEBIAN_FRONTEND=noninteractive apt install -y wget software-properties-common
30+
DEBIAN_FRONTEND=noninteractive apt-get clean all
31+
wget -qO - https://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
32+
if [[ $ROCM_DEB_REPO == https://repo.radeon.com/rocm/* ]]; then
33+
echo "deb [arch=amd64] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" >/etc/apt/sources.list.d/rocm.list
34+
else
35+
echo "deb [arch=amd64 trusted=yes] $ROCM_DEB_REPO $ROCM_BUILD_NAME $ROCM_BUILD_NUM" >/etc/apt/sources.list.d/rocm.list
36+
fi
37+
#Install rocm and other packages
38+
apt-get update --allow-insecure-repositories && DEBIAN_FRONTEND=noninteractive apt-get install -y \
39+
build-essential \
40+
software-properties-common \
41+
clang-6.0 \
42+
clang-format-6.0 \
43+
curl \
44+
g++-multilib \
45+
git \
46+
vim \
47+
libnuma-dev \
48+
virtualenv \
49+
python3-pip \
50+
pciutils \
51+
python-is-python3 \
52+
libffi-dev \
53+
libssl-dev \
54+
build-essential \
55+
zlib1g-dev \
56+
libbz2-dev \
57+
libreadline-dev \
58+
libsqlite3-dev curl \
59+
libncursesw5-dev \
60+
xz-utils \
61+
tk-dev \
62+
libxml2-dev \
63+
libxmlsec1-dev \
64+
libffi-dev \
65+
liblzma-dev \
66+
wget \
67+
rocm-dev \
68+
rocm-libs \
69+
miopen-hip \
70+
miopen-hip-dev \
71+
rocblas \
72+
rocblas-dev \
73+
rocsolver-dev \
74+
rocrand-dev \
75+
rocfft-dev \
76+
hipfft-dev \
77+
hipblas-dev \
78+
rocprim-dev \
79+
hipcub-dev \
80+
rccl-dev \
81+
hipsparse-dev \
82+
hipsolver-dev \
83+
wget &&
84+
apt-get clean &&
85+
rm -rf /var/lib/apt/lists/*
8786

8887
else
89-
bash "/${CUSTOM_INSTALL}"
88+
bash "/${CUSTOM_INSTALL}"
9089
fi
9190

9291
echo $ROCM_VERSION
@@ -95,6 +94,6 @@ echo $ROCM_PATH
9594
echo $GPU_DEVICE_TARGETS
9695

9796
# Ensure the ROCm target list is set up
98-
GPU_DEVICE_TARGETS=${GPU_DEVICE_TARGETS:-"gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1200 gfx1201"}
97+
GPU_DEVICE_TARGETS=${GPU_DEVICE_TARGETS:-"gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201"}
9998
printf '%s\n' ${GPU_DEVICE_TARGETS} | tee -a "$ROCM_PATH/bin/target.lst"
10099
touch "${ROCM_PATH}/.info/version"

‎build/rocm/tools/build_wheels.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@
3535
LOG = logging.getLogger(__name__)
3636

3737

38-
GPU_DEVICE_TARGETS = "gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1200 gfx1201"
38+
GPU_DEVICE_TARGETS = "gfx900 gfx906 gfx908 gfx90a gfx940 gfx941 gfx942 gfx1030 gfx1100 gfx1101 gfx1200 gfx1201"
3939

4040

4141
def build_rocm_path(rocm_version_str):

0 commit comments

Comments
 (0)