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
|
#!/usr/bin/env python
#
# Copyright 2018 Google Inc.
#
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Create the asset."""
import argparse
import common
import multiprocessing
import os
import shutil
import subprocess
import utils
def create_asset(target_dir):
"""Create the asset."""
# Check out and build the Intel NEO driver. Following instructions here:
# https://github.com/intel/compute-runtime/blob/master/documentation/BUILD_Ubuntu.md
with utils.tmp_dir():
# Install build deps.
neo_build_deps = ['ccache', 'flex', 'bison', 'clang-4.0', 'cmake', 'g++',
'git', 'patch', 'zlib1g-dev', 'autoconf', 'xutils-dev',
'libtool', 'pkg-config', 'libpciaccess-dev']
apt_get_cmd = ['sudo', 'apt-get', 'install'] + neo_build_deps
print 'Running "%s"' % ' '.join(apt_get_cmd)
subprocess.check_call(apt_get_cmd)
# Check out repos.
for [repo, branch, local_name] in [
['llvm-mirror/clang', 'release_40', 'clang_source'],
['intel/opencl-clang', 'master', 'common_clang'],
['intel/llvm-patches', 'master', 'llvm_patches'],
['llvm-mirror/llvm', 'release_40', 'llvm_source'],
['intel/gmmlib', 'master', 'gmmlib'],
['intel/intel-graphics-compiler', 'master', 'igc'],
['KhronosGroup/OpenCL-Headers', 'master', 'opencl_headers'],
['intel/compute-runtime', 'master', 'neo']
]:
subprocess.check_call(['git', 'clone', '--depth', '1', '--branch', branch,
'https://github.com/' + repo, local_name])
# Configure the build.
build_dir = os.path.join(os.getcwd(), 'build')
os.mkdir(build_dir)
os.chdir(build_dir)
subprocess.check_call(['cmake', '-DBUILD_TYPE=Release',
'-DCMAKE_BUILD_TYPE=Release', '../neo'])
# Build and package the library.
subprocess.check_call(['make', '-j%d' % multiprocessing.cpu_count(),
'package'])
# Extract library and move necessary files to target_dir. We ignore the ICD
# file because it's generated on the bot after we know the path to the CIPD
# package.
subprocess.check_call(['dpkg-deb', '--extract',
'intel-opencl-1.0-0.x86_64-igdrcl.deb', build_dir])
lib_dir = os.path.join(build_dir, 'usr', 'local', 'lib')
for f in os.listdir(lib_dir):
shutil.move(os.path.join(lib_dir, f), target_dir)
def main():
parser = argparse.ArgumentParser()
parser.add_argument('--target_dir', '-t', required=True)
args = parser.parse_args()
create_asset(args.target_dir)
if __name__ == '__main__':
main()
|
