forked from plaidml/plaidml
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path.travis.yml
33 lines (29 loc) · 1.21 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
language: python
matrix:
include:
- python: "2.7"
env:
- BZL_PY_VER=PY2
- PLAIDML_EXPERIMENTAL=1
- python: "3.6"
env:
- BZL_PY_VER=PY3
- PLAIDML_EXPERIMENTAL=1
before_install:
- sudo apt-get install openjdk-8-jdk bison g++ wget -y
- wget https://github.com/bazelbuild/bazel/releases/download/0.9.0/bazel-0.9.0-without-jdk-installer-linux-x86_64.sh
- chmod +x ./bazel-0.9.0-without-jdk-installer-linux-x86_64.sh
- sudo ./bazel-0.9.0-without-jdk-installer-linux-x86_64.sh
- wget https://storage.googleapis.com/external_build_repo/opencl_runtime_16.1.1_x64_ubuntu_6.4.0.25.tgz
- tar xzvf opencl_runtime_16.1.1_x64_ubuntu_6.4.0.25.tgz
- pushd opencl_runtime_16.1.1_x64_ubuntu_6.4.0.25
- sudo ./install.sh --cli-mode --silent ../tools/intel-ocl-silent
- popd
- pip install tensorflow theano setuptools h5py
- git clone http://github.com/plaidml/plaidbench
install:
- bazel build -c opt --force_python=$BZL_PY_VER plaidml:wheel plaidml/keras:wheel
- pip install bazel-bin/plaidml/*whl bazel-bin/plaidml/keras/*whl
script:
- bazel test -c opt --force_python=$BZL_PY_VER --test_env=PLAIDML_EXPERIMENTAL plaidml/keras:backend_test
- python plaidbench/plaidbench.py mobilenet