forked from ROCm/aws-ofi-rccl
-
Notifications
You must be signed in to change notification settings - Fork 0
/
.travis.yml
52 lines (44 loc) · 1.31 KB
/
.travis.yml
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
language: c
sudo: true
dist: xenial
env:
global:
- PATH=/usr/local/cuda/bin:${PATH}
- LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
addons:
apt:
packages:
- openmpi-bin
- openmpi-common
- libopenmpi-dev
- gcc
- automake
- autoconf
- libtool
- flex
before_install:
# Install CUDA
- wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64
- sudo dpkg -i cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64
- sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
- sudo apt-get -y update
- rm -f cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64
- sudo apt-get install -y cuda
install:
# Install libfabric
- git clone https://github.com/ofiwg/libfabric.git
- pushd libfabric
- ./autogen.sh
- ./configure --enable-debug --disable-psm3; make -j8; sudo make install
- popd
# Install NCCL
- git clone https://github.com/NVIDIA/nccl.git
- pushd nccl
- make src.build NVCC_GENCODE="-gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_60,code=sm_60"
- popd
before_script: sudo ldconfig
script:
# Build aws-ofi-nccl plugin
- ./autogen.sh
- ./configure --with-nccl=${PWD}/nccl/build --with-cuda=/usr/local/cuda
- make && sudo make install