diff --git a/dockerfile/cuda11.1.1.dockerfile b/dockerfile/cuda11.1.1.dockerfile index 6300eb4bd..65df9d6e5 100644 --- a/dockerfile/cuda11.1.1.dockerfile +++ b/dockerfile/cuda11.1.1.dockerfile @@ -13,7 +13,7 @@ FROM nvcr.io/nvidia/pytorch:20.12-py3 # - HPC-X: v2.8.3 # - NCCL RDMA SHARP plugins: 7cccbc1 # Intel: -# - mlc: v3.10 +# - mlc: v3.11 LABEL maintainer="SuperBench" @@ -112,7 +112,7 @@ RUN cd /tmp && \ # Install Intel MLC RUN cd /tmp && \ - wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \ + wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \ tar xzf mlc.tgz Linux/mlc && \ cp ./Linux/mlc /usr/local/bin/ && \ rm -rf ./Linux mlc.tgz diff --git a/dockerfile/cuda12.2.dockerfile b/dockerfile/cuda12.2.dockerfile index 32d28bbcc..9c58c0eca 100644 --- a/dockerfile/cuda12.2.dockerfile +++ b/dockerfile/cuda12.2.dockerfile @@ -12,7 +12,7 @@ FROM nvcr.io/nvidia/pytorch:23.10-py3 # - OFED: 23.07-0.5.1.2 # - HPC-X: v2.16 # Intel: -# - mlc: v3.10 +# - mlc: v3.11 LABEL maintainer="SuperBench" @@ -95,7 +95,7 @@ RUN cd /opt && \ # Install Intel MLC RUN cd /tmp && \ - wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \ + wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \ tar xzf mlc.tgz Linux/mlc && \ cp ./Linux/mlc /usr/local/bin/ && \ rm -rf ./Linux mlc.tgz diff --git a/dockerfile/rocm5.0.x.dockerfile b/dockerfile/rocm5.0.x.dockerfile index d9cdeb307..8bb718794 100644 --- a/dockerfile/rocm5.0.x.dockerfile +++ b/dockerfile/rocm5.0.x.dockerfile @@ -17,7 +17,7 @@ FROM ${BASE_IMAGE} # Mellanox: # - OFED: 5.2-2.2.3.0 # Intel: -# - mlc: v3.10 +# - mlc: v3.11 LABEL maintainer="SuperBench" @@ -98,7 +98,7 @@ RUN cd /tmp && \ # Install Intel MLC RUN cd /tmp && \ - wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \ + wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \ tar xzf mlc.tgz Linux/mlc && \ cp ./Linux/mlc /usr/local/bin/ && \ rm -rf ./Linux mlc.tgz diff --git a/dockerfile/rocm5.1.x.dockerfile b/dockerfile/rocm5.1.x.dockerfile index 8401935d6..e56c35c27 100644 --- a/dockerfile/rocm5.1.x.dockerfile +++ b/dockerfile/rocm5.1.x.dockerfile @@ -16,7 +16,7 @@ FROM ${BASE_IMAGE} # Mellanox: # - OFED: 5.2-2.2.3.0 # Intel: -# - mlc: v3.10 +# - mlc: v3.11 LABEL maintainer="SuperBench" @@ -110,7 +110,7 @@ RUN cd /tmp && \ # Install Intel MLC RUN cd /tmp && \ - wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \ + wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \ tar xzf mlc.tgz Linux/mlc && \ cp ./Linux/mlc /usr/local/bin/ && \ rm -rf ./Linux mlc.tgz @@ -142,7 +142,7 @@ RUN echo PATH="$PATH" > /etc/environment && \ WORKDIR ${SB_HOME} ADD third_party third_party -RUN make ROCBLAS_BRANCH=release/rocm-rel-5.1 -C third_party rocm -o rocm_hipblaslt -o megatron_deepspeed -o megatron_lm +RUN make ROCBLAS_BRANCH=release/rocm-rel-5.1 -C third_party rocm -o rocm_hipblaslt -o megatron_deepspeed -o megatron_lm ADD . . RUN python3 -m pip install --no-cache-dir .[amdworker] && \ diff --git a/dockerfile/rocm5.7.x.dockerfile b/dockerfile/rocm5.7.x.dockerfile index 4d68073cc..87af0e368 100644 --- a/dockerfile/rocm5.7.x.dockerfile +++ b/dockerfile/rocm5.7.x.dockerfile @@ -8,7 +8,7 @@ FROM ${BASE_IMAGE} # ROCm: # - ROCm: 5.7 # Intel: -# - mlc: v3.10 +# - mlc: v3.11 LABEL maintainer="SuperBench" @@ -126,7 +126,7 @@ RUN cd /tmp && \ # Install Intel MLC RUN cd /tmp && \ - wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \ + wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \ tar xzf mlc.tgz Linux/mlc && \ cp ./Linux/mlc /usr/local/bin/ && \ rm -rf ./Linux mlc.tgz diff --git a/dockerfile/rocm6.0.x.dockerfile b/dockerfile/rocm6.0.x.dockerfile index 57194575e..648bc195d 100644 --- a/dockerfile/rocm6.0.x.dockerfile +++ b/dockerfile/rocm6.0.x.dockerfile @@ -14,7 +14,7 @@ FROM ${BASE_IMAGE} # - openmpi: 4.1.x # - apex: 1.0.0 # Intel: -# - mlc: v3.10 +# - mlc: v3.11 LABEL maintainer="SuperBench" @@ -132,7 +132,7 @@ RUN cd /tmp && \ # Install Intel MLC RUN cd /tmp && \ - wget -q https://downloadmirror.intel.com/763324/mlc_v3.10.tgz -O mlc.tgz && \ + wget -q https://downloadmirror.intel.com/793041/mlc_v3.11.tgz -O mlc.tgz && \ tar xzf mlc.tgz Linux/mlc && \ cp ./Linux/mlc /usr/local/bin/ && \ rm -rf ./Linux mlc.tgz