Skip to content

Commit 5d4290a

Browse files
committed
update cupy to latest commit with rocm 6.0 fixes
1 parent d876d37 commit 5d4290a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tools/ci_build/github/pai/rocm-ci-pipeline-env.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ RUN wget https://download.open-mpi.org/release/open-mpi/v4.1/openmpi-4.1.5.tar.b
9696
# Install CuPy, No stable version is available
9797
RUN pip uninstall -y cython
9898
RUN git clone https://github.com/ROCmSoftwarePlatform/cupy && cd cupy && \
99-
git checkout fc251a808037f8a2270860c2a23a683bfc0de43e && \
99+
git checkout 432a8683351d681e00903640489cb2f4055d2e09 && \
100100
export CUPY_INSTALL_USE_HIP=1 && \
101101
export ROCM_HOME=/opt/rocm && \
102102
export HCC_AMDGPU_TARGET=gfx906,gfx908,gfx90a && \

0 commit comments

Comments
 (0)