Add compatability wrapper for pyopencl.

This fixes pyopencl caching when executing in a bazel sandbox.

github.com//issues/65

datasets/benchmarks/jacobi_opencl/jacobi_opencl.py

@@ -35,9 +35,9 @@
 import typing
 
 import numpy as np
-import pyopencl as CL
 
 from labm8.py import app
+from third_party.py.pyopencl import pyopencl as CL
 
 FLAGS = app.FLAGS
 

experimental/dsmith/opencl/db.py

@@ -603,7 +603,7 @@ def __repr__(self) -> str:
 
   def platform_id(self):
     """ return OpenCL platform index, or KeyError if platform not found """
-    import pyopencl as cl
+    from third_party.py.pyopencl import pyopencl as cl
 
     for i, platform in enumerate(cl.get_platforms()):
       if platform.get_info(cl.platform_info.NAME) == self.platform:
@@ -612,7 +612,7 @@ def platform_id(self):
 
   def device_id(self):
     """ return OpenCL device index, or KeyError if device not found """
-    import pyopencl as cl
+    from third_party.py.pyopencl import pyopencl as cl
 
     platform = cl.get_platforms()[self.platform_id()]
     ctx = cl.Context(properties=[(cl.context_properties.PLATFORM, platform)])
@@ -700,7 +700,7 @@ def from_env(
 
   @staticmethod
   def _get_ids(platform: str, device: str, driver: str) -> Tuple[int, int]:
-    import pyopencl as cl
+    from third_party.py.pyopencl import pyopencl as cl
 
     # match platform ID:
     for j, cl_platform in enumerate(cl.get_platforms()):

experimental/dsmith/scrapheap/clgen_run_cl_launcher.py

@@ -6,12 +6,13 @@
 from typing import Tuple
 
 import progressbar
-import pyopencl as cl
 from dsmith import clsmith
 from dsmith import db
 from dsmith.db import *
 from dsmith.lib import *
 
+from third_party.py.pyopencl import pyopencl as cl
+
 
 def get_platform_name(platform_id):
   platform = cl.get_platforms()[platform_id]

experimental/dsmith/scrapheap/clsmith_run_cl_launcher.py

@@ -8,13 +8,13 @@
 from typing import Tuple
 
 import progressbar
-import pyopencl as cl
 from dsmith import clsmith
 from dsmith import db
 from dsmith.db import *
 from dsmith.lib import *
 
 from labm8.py import crypto
+from third_party.py.pyopencl import pyopencl as cl
 
 
 def get_platform_name(platform_id):

experimental/dsmith/scrapheap/run_programs.py

@@ -3,10 +3,11 @@
 from itertools import product
 from subprocess import Popen
 
-import pyopencl as cl
 from dsmith import db
 from dsmith.db import *
 
+from third_party.py.pyopencl import pyopencl as cl
+
 
 def get_platform_name(platform_id):
   platform = cl.get_platforms()[platform_id]

experimental/dsmith/scrapheap/run_reductions.py

@@ -2,11 +2,11 @@
 from collections import namedtuple
 from typing import NewType
 
-import pyopencl as cl
 from dsmith.db import *
 from dsmith.lib import *
 
 from labm8.py import fs
+from third_party.py.pyopencl import pyopencl as cl
 
 # paths to clreduce library
 CLREDUCE_DIR = fs.abspath("..", "lib", "clreduce")

experimental/quicksilver/jacobi_demo.py

@@ -62,12 +62,11 @@
 """
 import time
 
-import pyopencl as CL
-
 from datasets.benchmarks.jacobi_opencl import jacobi_opencl as jacobi
 from labm8.py import app
 from labm8.py import humanize
 from labm8.py import prof
+from third_party.py.pyopencl import pyopencl as CL
 
 FLAGS = app.FLAGS
 

gpu/omnitune/omnitune/opencl.py

@@ -1,4 +1,4 @@
-import pyopencl as cl
+from third_party.py.pyopencl import pyopencl as cl
 
 
 def get_devices():

third_party/py/pyopencl/BUILD

@@ -8,8 +8,9 @@ licenses(["notice"])  # MIT
 
 py_library(
     name = "pyopencl",
-    srcs = ["//third_party/py:empty.py"],
+    srcs = ["pyopencl.py"],
     deps = [
         requirement("pyopencl"),
+        requirement("pytools"),  # Implicit dependency.
     ],
 )

third_party/py/pyopencl/pyopencl.py

@@ -0,0 +1,15 @@
+"""pyopencl module shim.
+
+This is a drop-in replacement for pyopencl that enables support for execution
+inside of bazel sandboxing.
+"""
+import os
+
+# Switch out the default value for ~/.cache on Linux when executing in
+# bazel sandbox. See:
+# https://docs.bazel.build/versions/master/test-encyclopedia.html#initial-conditions
+_BAZEL_TEST_TMPDIR = os.environ.get("TEST_TMPDIR")
+if _BAZEL_TEST_TMPDIR:
+  os.environ["XDG_CACHE_HOME"] = _BAZEL_TEST_TMPDIR
+
+from pyopencl import *