def test_toolchain(self):
"""Toolchain"""
mv2 = mvapich2_gdr()
tc = mv2.toolchain
self.assertEqual(tc.CC, 'mpicc')
self.assertEqual(tc.CXX, 'mpicxx')
self.assertEqual(tc.FC, 'mpifort')
self.assertEqual(tc.F77, 'mpif77')
self.assertEqual(tc.F90, 'mpif90')
def test_runtime(self):
"""Runtime"""
mv2 = mvapich2_gdr()
r = mv2.runtime()
s = '\n'.join(str(x) for x in r)
self.assertEqual(s,
r'''# MVAPICH2-GDR
RUN apt-get update -y && \
apt-get install -y --no-install-recommends \
openssh-client && \
rm -rf /var/lib/apt/lists/*
COPY --from=0 /opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5 /opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5
ENV LD_LIBRARY_PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5/lib64:$LD_LIBRARY_PATH \
MV2_USE_GPUDIRECT=0 \
MV2_USE_GPUDIRECT_GDRCOPY=0 \
PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5/bin:$PATH''')
def test_default_centos(self):
"""Default mvapich2_gdr building block"""
mv2 = mvapich2_gdr()
self.assertEqual(str(mv2),
r'''# MVAPICH2-GDR version 2.3a
RUN yum install -y \
openssh-clients \
perl \
wget && \
rm -rf /var/cache/yum/*
RUN mkdir -p /tmp && wget -q --no-check-certificate -P /tmp http://mvapich.cse.ohio-state.edu/download/mvapich/gdr/2.3a/mofed3.4/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5-2.3a-1.el7.centos.x86_64.rpm && \
rpm --install --nodeps /tmp/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5-2.3a-1.el7.centos.x86_64.rpm && \
(test -f /usr/bin/bash || ln -s /bin/bash /usr/bin/bash) && \
ln -s /usr/local/cuda/lib64/stubs/nvidia-ml.so /usr/local/cuda/lib64/stubs/nvidia-ml.so.1 && \
rm -rf /tmp/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5-2.3a-1.el7.centos.x86_64.rpm
ENV LD_LIBRARY_PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5/lib64:$LD_LIBRARY_PATH \
MV2_USE_GPUDIRECT=0 \
MV2_USE_GPUDIRECT_GDRCOPY=0 \
PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5/bin:$PATH \
PROFILE_POSTLIB="-L/usr/local/cuda/lib64/stubs -lnvidia-ml"''')
def test_defaults_ubuntu(self):
"""Default mvapich2_gdr building block"""
mv2 = mvapich2_gdr()
self.assertEqual(str(mv2),
r'''# MVAPICH2-GDR version 2.3a
RUN apt-get update -y && \
apt-get install -y --no-install-recommends \
openssh-client \
wget && \
rm -rf /var/lib/apt/lists/*
RUN mkdir -p /tmp && wget -q --no-check-certificate -P /tmp http://mvapich.cse.ohio-state.edu/download/mvapich/gdr/2.3a/mofed3.4/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5_2.3a-1.el7.centos_amd64.deb && \
dpkg --install /tmp/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5_2.3a-1.el7.centos_amd64.deb && \
(test -f /usr/bin/bash || ln -s /bin/bash /usr/bin/bash) && \
ln -s /usr/local/cuda/lib64/stubs/nvidia-ml.so /usr/local/cuda/lib64/stubs/nvidia-ml.so.1 && \
rm -rf /tmp/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5_2.3a-1.el7.centos_amd64.deb
ENV LD_LIBRARY_PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5/lib64:$LD_LIBRARY_PATH \
MV2_USE_GPUDIRECT=0 \
MV2_USE_GPUDIRECT_GDRCOPY=0 \
PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5/bin:$PATH \
PROFILE_POSTLIB="-L/usr/local/cuda/lib64/stubs -lnvidia-ml"''')
def test_package(self):
"""Directory in local build context"""
mv2 = mvapich2_gdr(package='mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5_2.3a-1.el7.centos_amd64.deb')
self.assertEqual(str(mv2),
r'''# MVAPICH2-GDR
RUN apt-get update -y && \
apt-get install -y --no-install-recommends \
openssh-client \
wget && \
rm -rf /var/lib/apt/lists/*
COPY mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5_2.3a-1.el7.centos_amd64.deb /tmp/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5_2.3a-1.el7.centos_amd64.deb
RUN dpkg --install /tmp/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5_2.3a-1.el7.centos_amd64.deb && \
(test -f /usr/bin/bash || ln -s /bin/bash /usr/bin/bash) && \
ln -s /usr/local/cuda/lib64/stubs/nvidia-ml.so /usr/local/cuda/lib64/stubs/nvidia-ml.so.1 && \
rm -rf /tmp/mvapich2-gdr-mcast.cuda9.0.mofed3.4.gnu4.8.5_2.3a-1.el7.centos_amd64.deb
ENV LD_LIBRARY_PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5/lib64:$LD_LIBRARY_PATH \
MV2_USE_GPUDIRECT=0 \
MV2_USE_GPUDIRECT_GDRCOPY=0 \
PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda9.0/mofed3.4/mpirun/gnu4.8.5/bin:$PATH \
PROFILE_POSTLIB="-L/usr/local/cuda/lib64/stubs -lnvidia-ml"''')
def test_options(self):
"""PGI compiler and different Mellanox OFED version"""
mv2 = mvapich2_gdr(pgi=True, gnu=False, cuda_version='8.0',
mlnx_ofed_version='4.0')
self.assertEqual(str(mv2),
r'''# MVAPICH2-GDR version 2.3a
RUN apt-get update -y && \
apt-get install -y --no-install-recommends \
openssh-client \
wget && \
rm -rf /var/lib/apt/lists/*
RUN mkdir -p /tmp && wget -q --no-check-certificate -P /tmp http://mvapich.cse.ohio-state.edu/download/mvapich/gdr/2.3a/mofed4.0/mvapich2-gdr-mcast.cuda8.0.mofed4.0.pgi17.10_2.3a-1.el7.centos_amd64.deb && \
dpkg --install /tmp/mvapich2-gdr-mcast.cuda8.0.mofed4.0.pgi17.10_2.3a-1.el7.centos_amd64.deb && \
(test -f /usr/bin/bash || ln -s /bin/bash /usr/bin/bash) && \
ln -s /usr/local/cuda/lib64/stubs/nvidia-ml.so /usr/local/cuda/lib64/stubs/nvidia-ml.so.1 && \
rm -rf /tmp/mvapich2-gdr-mcast.cuda8.0.mofed4.0.pgi17.10_2.3a-1.el7.centos_amd64.deb
ENV LD_LIBRARY_PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda8.0/mofed4.0/mpirun/pgi17.10/lib64:$LD_LIBRARY_PATH \
MV2_USE_GPUDIRECT=0 \
MV2_USE_GPUDIRECT_GDRCOPY=0 \
PATH=/opt/mvapich2/gdr/2.3a/mcast/no-openacc/cuda8.0/mofed4.0/mpirun/pgi17.10/bin:$PATH \
PROFILE_POSTLIB="-L/usr/local/cuda/lib64/stubs -lnvidia-ml"''')
