Skip to content
Snippets Groups Projects
Commit f6985106 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Don't import pyopencl at the top level in pyopencl target

parent 17785183
No related branches found
No related tags found
No related merge requests found
Pipeline #
...@@ -31,12 +31,6 @@ import numpy as np ...@@ -31,12 +31,6 @@ import numpy as np
from loopy.target.opencl import OpenCLTarget from loopy.target.opencl import OpenCLTarget
import pyopencl as cl
import pyopencl.characterize as cl_char
# This ensures the dtype registry is populated.
import pyopencl.tools # noqa
import logging import logging
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
...@@ -44,6 +38,9 @@ logger = logging.getLogger(__name__) ...@@ -44,6 +38,9 @@ logger = logging.getLogger(__name__)
# {{{ temp storage adjust for bank conflict # {{{ temp storage adjust for bank conflict
def adjust_local_temp_var_storage(kernel, device): def adjust_local_temp_var_storage(kernel, device):
import pyopencl as cl
import pyopencl.characterize as cl_char
logger.debug("%s: adjust temp var storage" % kernel.name) logger.debug("%s: adjust temp var storage" % kernel.name)
new_temp_vars = {} new_temp_vars = {}
...@@ -256,6 +253,9 @@ class PyOpenCLTarget(OpenCLTarget): ...@@ -256,6 +253,9 @@ class PyOpenCLTarget(OpenCLTarget):
""" """
def __init__(self, device=None): def __init__(self, device=None):
# This ensures the dtype registry is populated.
import pyopencl.tools # noqa
super(PyOpenCLTarget, self).__init__() super(PyOpenCLTarget, self).__init__()
self.device = device self.device = device
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment