Skip to content

Commit 6c7b50f

Browse files
tianleiwuashrit-ms
authored andcommitted
[ROCm] Update CI to use rocm 6.3.2 (#23577)
### Description * Update rocm to 6.3.2; * Remove dependency on cupy (which does not support rocm 6.3 yet). ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. -->
1 parent f3e5de7 commit 6c7b50f

File tree

2 files changed

+3
-13
lines changed

2 files changed

+3
-13
lines changed

tools/ci_build/github/azure-pipelines/linux-rocm-ci-pipeline.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ variables:
3737
- name: render
3838
value: 109
3939
- name: RocmVersion
40-
value: 6.1.3
40+
value: 6.3.2
4141

4242
jobs:
4343
- job: Linux_Build
@@ -223,7 +223,7 @@ jobs:
223223
-e MKL_NUM_THREADS=1 \
224224
-e KERNEL_EXPLORER_BUILD_DIR=/build/Release \
225225
-e KERNEL_EXPLORER_BATCHED_GEMM_MAX_BATCH_SIZE=8 \
226-
-e KERNEL_EXPLORER_TEST_USE_CUPY=1 \
226+
-e KERNEL_EXPLORER_TEST_USE_CUPY=0 \
227227
-e CUPY_CACHE_DIR=/build/Release \
228228
onnxruntimerocm-cibuild-rocm$(RocmVersion) \
229229
/bin/bash -c "

tools/ci_build/github/linux/docker/rocm-ci-pipeline-env.Dockerfile

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
# Refer to https://github.com/RadeonOpenCompute/ROCm-docker/blob/master/dev/Dockerfile-ubuntu-22.04-complete
22
FROM ubuntu:22.04
33

4-
ARG ROCM_VERSION=6.1.3
4+
ARG ROCM_VERSION=6.3.2
55
ARG AMDGPU_VERSION=${ROCM_VERSION}
66
ARG APT_PREF='Package: *\nPin: release o=repo.radeon.com\nPin-Priority: 600'
77

@@ -88,13 +88,3 @@ RUN pip install packaging \
8888
numpy==1.26.4
8989

9090
RUN apt install -y git
91-
92-
# Install Cupy to decrease CPU utilization
93-
# Note that the version of Cupy requires numpy < 1.27
94-
RUN git clone https://github.com/ROCm/cupy && cd cupy && \
95-
git checkout 432a8683351d681e00903640489cb2f4055d2e09 && \
96-
export CUPY_INSTALL_USE_HIP=1 && \
97-
export ROCM_HOME=/opt/rocm && \
98-
export HCC_AMDGPU_TARGET=gfx906,gfx908,gfx90a && \
99-
git submodule update --init && \
100-
pip install -e . --no-cache-dir -vvvv

0 commit comments

Comments
 (0)