1 files changed, 47 insertions, 0 deletions
diff --git a/infra/bots/assets/opencl_ocl_icd_linux/create.py b/infra/bots/assets/opencl_ocl_icd_linux/create.py
new file mode 100755
index 0000000000..4b2512cfb8
--- /dev/null
+++ b/infra/bots/assets/opencl_ocl_icd_linux/create.py
@@ -0,0 +1,47 @@
+#!/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 os
+import shutil
+import subprocess
+import utils
+
+# Use libOpenCL.so from the ocl-icd-opencl-dev Debian package.
+PKGS = [
+    'ocl-icd-opencl-dev',
+    'ocl-icd-libopencl1',
+]
+
+def create_asset(target_dir):
+  """Create the asset."""
+  with utils.tmp_dir():
+    # Download required Debian packages.
+    subprocess.check_call(['apt-get', 'download'] + PKGS)
+    # Extract to CWD.
+    for f in os.listdir('.'):
+      subprocess.check_call(['dpkg-deb', '--extract', f, '.'])
+    # Copy usr/lib/x86_64-linux-gnu/* to target_dir.
+    lib_dir = os.path.join(os.getcwd(), 'usr', 'lib', 'x86_64-linux-gnu')
+    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()