-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathDockerfile
54 lines (41 loc) · 1.48 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
ARG IMAGE="rocm_gpu:6.3"
FROM ${IMAGE}
ARG JSON_FORTRAN_BRANCH="master"
ARG NEKO_BRANCH="v0.9.0"
# Update and Install basic Linux development tools
RUN apt-get update \
&& DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
libopenblas-dev \
&& apt-get clean
WORKDIR /tmp
RUN git clone -b ${JSON_FORTRAN_BRANCH} https://github.com/jacobwilliams/json-fortran.git \
&& cd json-fortran \
&& mkdir build \
&& cd build \
&& cmake -DCMAKE_INSTALL_PREFIX=/opt/json-fortran -DUSE_GNU_INSTALL_CONVENTION=ON .. \
&& make install
ENV PKG_CONFIG_PATH=/opt/json-fortran/lib/pkgconfig:$PKG_CONFIG_PATH \
LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/opt/json-fortran/lib
WORKDIR /opt
RUN git clone https://github.com/Nek5000/gslib.git \
&& cd gslib \
&& make
RUN git clone -b ${NEKO_BRANCH} https://github.com/ExtremeFLOW/neko \
&& cd neko \
&& ./regen.sh \
&& ./configure \
HIP_HIPCC_FLAGS="-O3 -fPIE --offload-arch=${GPU_TARGET}" \
HIPCC=${ROCM_PATH}/bin/hipcc \
--prefix=/opt/neko \
--with-hip=${ROCM_PATH} \
--enable-device-mpi \
--with-gslib=/opt/gslib/build/ \
&& make \
&& make install \
&& chmod -R 777 /opt/neko \
&& ln -s /opt/neko/examples/tgv/32768.nmsh /opt/neko/bench/tgv32/32768.nmsh
WORKDIR /opt/neko/bench
ENV PATH=$PATH:/opt/neko/bin \
PKG_CONFIG_PATH=$PKG_CONFIG_PATH:/opt/neko/lib/pkgconfig \
LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/opt/neko/lib
CMD ["/bin/bash"]