Add Starpu-HIP-CUDA

dockerfile-distrib

@@ -57,6 +57,8 @@ RUN update-alternatives --set libblas.so.3-x86_64-linux-gnu /usr/lib/x86_64-linu
 RUN update-alternatives --set liblapack.so-x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/lapack/liblapack.so
 RUN update-alternatives --set liblapack.so.3-x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/lapack/liblapack.so.3
 
+RUN mkdir -p /usr/local/cuda/bin/ /usr/local/cuda/include /usr/local/cuda/lib && ln -s /usr/bin/nvcc /usr/local/cuda/bin/nvcc
+
 # change the default shell to be bash
 SHELL ["/bin/bash", "-c"]
 
@@ -138,9 +140,39 @@ RUN cd $HOME && \
     cd starpu/ && \
     ./autogen.sh && \
     ./configure --prefix=/home/gitlab/install/starpu-hip --enable-debug --disable-cuda --enable-hip && \
-    make -j5 && \
-    make install
+    make -j5 install && \
+    cd $HOME && rm starpu/ -rf
+
+# Install Starpu (master) with HIP+CUDA
+ENV HIPCUDA_DIR=/home/gitlab/hipcuda
+ENV HIP_PLATFORM=nvidia
+ENV ROCM_BRANCH=rocm-5.4.x
+ENV PATH=$HIPCUDA_DIR/bin:$PATH
+ENV LD_LIBRARY_PATH=$HIPCUDA_DIR/lib
+RUN mkdir $HIPCUDA_DIR && \
+    cd $HOME && \
+    git clone -b "$ROCM_BRANCH" https://github.com/ROCm-Developer-Tools/hip.git && \
+    git clone -b "$ROCM_BRANCH" https://github.com/ROCm-Developer-Tools/hipamd.git && \
+    export HIP_DIR="$(readlink -f hip)" && export HIPAMD_DIR="$(readlink -f hipamd)" && \
+    cd "$HIPAMD_DIR" && \
+    mkdir -p build; cd build && \
+    cmake .. -DHIP_COMMON_DIR=$HIP_DIR -DHIP_PLATFORM=nvidia -DCMAKE_INSTALL_PREFIX=$HIPCUDA_DIR && \
+    make -j5 install
+RUN cd $HOME && \
+    git clone https://github.com/ROCmSoftwarePlatform/hipBLAS.git && \
+    cd hipBLAS/ && mkdir build && cd build && \
+    cmake .. -DCMAKE_INSTALL_PREFIX=$HIPCUDA_DIR -DHIP_ROOT_DIR=$HIPCUDA_DIR -DUSE_CUDA=ON && \
+    make -j5 install
+RUN cd $HOME && \
+    git clone https://gitlab.inria.fr/starpu/starpu.git && \
+    cd starpu/ && \
+    ./autogen.sh && \
+    ./configure --prefix=/home/gitlab/install/starpu-hipcuda --enable-debug --disable-cuda --enable-hip --with-hipblas=$HIPCUDA_DIR && \
+    make -j5 install && \
+    cd $HOME && rm starpu/ -rf
+ENV PATH=/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
 ENV LD_LIBRARY_PATH=""
+ENV HIP_PLATFORM=""
 
 # Install Chameleon
 RUN cd $HOME && \