first commit

ci/build.sh

@@ -0,0 +1,19 @@
+set -x -e
+
+source ci/env.sh
+
+which g++
+which nvcc
+which cmake
+
+g++ --version
+nvcc --version
+cmake --version
+
+mkdir build
+cd build
+cmake .. \
+  -DCMAKE_BUILD_TYPE=$BUILD_TYPE \
+  -DCMAKE_PREFIX_PATH=$OPENMPI_PATH \
+  -DMPI_HOME=/usr/lib/x86_64-linux-gnu/openmpi/
+make VERBOSE=1 

ci/env.sh

@@ -0,0 +1,8 @@
+OPENMPI_PREFIX=$HOME/openmpi
+CMAKE_PREFIX=$HOME/cmake
+
+export PATH=/usr/local/cuda/bin:$PATH
+export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
+
+export PATH=$OPENMPI_PREFIX/bin:$PATH
+export PATH=$CMAKE_PREFIX/bin:$PATH

ci/install_deps.sh

@@ -0,0 +1,40 @@
+set -x -e
+
+source ci/env.sh
+
+# Install Cmake if it doesn't exist
+mkdir -p $CMAKE_PREFIX
+if [[ ! -f $CMAKE_PREFIX/bin/cmake ]]; then
+    if [[ $TRAVIS_CPU_ARCH == "ppc64le" ]]; then
+        wget -qSL https://cmake.org/files/v3.13/cmake-3.13.5.tar.gz -O cmake.tar.gz
+        tar -xf cmake.tar.gz --strip-components=1 -C $CMAKE_PREFIX
+        rm cmake.tar.gz
+        cd $CMAKE_PREFIX
+        ./bootstrap --prefix=$CMAKE_PREFIX
+        make -j `nproc` install
+    elif [[ $TRAVIS_CPU_ARCH == "amd64" ]]; then
+        wget -qSL https://cmake.org/files/v3.13/cmake-3.13.5-Linux-x86_64.tar.gz -O cmake.tar.gz
+        tar -xf cmake.tar.gz --strip-components=1 -C $CMAKE_PREFIX
+        rm cmake.tar.gz
+    fi
+fi
+cd $HOME
+
+sudo apt-get update 
+sudo apt-get install -y --no-install-recommends \
+  libopenmpi-dev openmpi-bin
+
+## install CUDA
+sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
+
+if [[ $TRAVIS_CPU_ARCH == "ppc64le" ]]; then
+    CUDA102="https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/ppc64el/cuda-repo-ubuntu1804_10.2.89-1_ppc64el.deb"
+elif [[ $TRAVIS_CPU_ARCH == "amd64" ]]; then
+    CUDA102="http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.2.89-1_amd64.deb"
+fi
+
+wget -SL $CUDA102 -O cuda.deb
+sudo dpkg -i cuda.deb
+sudo apt-get update 
+sudo apt-get install -y --no-install-recommends \
+  cuda-toolkit-10-2

ci/test.sh

@@ -0,0 +1,4 @@
+set -x
+
+cd build
+make test