-
Notifications
You must be signed in to change notification settings - Fork 10
/
.travis.yml
115 lines (102 loc) · 3.68 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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
#=========================================================================
# TravisCI Setup
#=========================================================================
#------------------------------------------------------------------------
# language and build matrix
#------------------------------------------------------------------------
language: cpp
git:
- depth: 3
python:
- 3.6
dist: bionic
env:
- CONFIG=""
compiler:
- gcc
cache: ccache
#------------------------------------------------------------------------
# install dependencies
#------------------------------------------------------------------------
before_install:
- pyenv global 3.6
install:
# Install Python requirements
- pip install --upgrade pip setuptools
- pip install numpy pyyaml mkl mkl-include setuptools cmake cffi typing sklearn tqdm pytest hypothesis
- pip list
#------------------------------------------------------------------------
# Build and run pytorch tests with emulation layer
#------------------------------------------------------------------------
jobs:
include:
- stage: cache warmup
script:
# Disable ASLR
- sudo sh -c "echo 0 > /proc/sys/kernel/randomize_va_space"
- cat /proc/sys/kernel/randomize_va_space
# Enable compiler cache
- sudo ln -s /usr/lib/ccache /usr/local/bin/gcc
- sudo ln -s /usr/lib/ccache /usr/local/bin/g++
# Check gcc version
- ldd --version
- gcc --version
# Env setup
- export MAX_JOBS=2
- export BUILD_TEST=0
- export USE_CUDA=0
- export USE_CUDNN=0
- export USE_FBGEMM=0
- export USE_MKL=0
- export USE_MKLDNN=0
- export USE_NNPACK=0
- export USE_QNNPACK=0
- export USE_DISTRIBUTED=0
- export USE_OPENMP=0
- export ATEN_THREADING=NATIVE
# Disable GCC warnings
- export CFLAGS="-w"
# Enable emulation layer
- export USE_HB_EMUL=1
# Build pytorch
- python setup.py build --cmake-only
- cd build
- make torch -j2
- stage: build pytorch
script:
# Disable ASLR
- sudo sh -c "echo 0 > /proc/sys/kernel/randomize_va_space"
- cat /proc/sys/kernel/randomize_va_space
# Enable compiler cache
- sudo ln -s /usr/lib/ccache /usr/local/bin/gcc
- sudo ln -s /usr/lib/ccache /usr/local/bin/g++
# Check gcc version
- ldd --version
- gcc --version
# Env setup
- export MAX_JOBS=2
- export BUILD_TEST=0
- export USE_CUDA=0
- export USE_CUDNN=0
- export USE_FBGEMM=0
- export USE_MKL=0
- export USE_MKLDNN=0
- export USE_NNPACK=0
- export USE_QNNPACK=0
- export USE_DISTRIBUTED=0
- export USE_OPENMP=0
- export ATEN_THREADING=NATIVE
# Disable GCC warnings
- export CFLAGS="-w"
# Enable emulation layer
- export USE_HB_EMUL=1
# Mutlitile execution
- export HBEMUL_TILE_X_DIM=16
- export HBEMUL_TILE_Y_DIM=8
# Build and install pytorch
- python setup.py develop
# Check torch
- pip list
# Run pytests
- cd hammerblade/torch/
- python pytest_runner.py