From e2ba90140834cea562c55d4d268a213c0756e9c7 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Mon, 18 Jan 2021 16:18:08 -0600
Subject: [PATCH] Invoker generator: globalize some py_codegen imports
---
pyopencl/invoker.py | 8 +-------
1 file changed, 1 insertion(+), 7 deletions(-)
diff --git a/pyopencl/invoker.py b/pyopencl/invoker.py
index c7fe8560..3feb7685 100644
--- a/pyopencl/invoker.py
+++ b/pyopencl/invoker.py
@@ -28,7 +28,7 @@ import numpy as np
from warnings import warn
import pyopencl._cl as _cl
from pytools.persistent_dict import WriteOncePersistentDict
-from pytools.py_codegen import Indentation
+from pytools.py_codegen import Indentation, PythonCodeGenerator
from pyopencl.tools import _NumpyTypesKeyBuilder
_PYPY = "__pypy__" in sys.builtin_module_names
@@ -54,7 +54,6 @@ del _size_t_char
# {{{ generic arg handling body
def generate_generic_arg_handling_body(num_args):
- from pytools.py_codegen import PythonCodeGenerator
gen = PythonCodeGenerator()
if num_args == 0:
@@ -85,7 +84,6 @@ def generate_specific_arg_handling_body(function_name,
fp_arg_count = 0
cl_arg_idx = 0
- from pytools.py_codegen import PythonCodeGenerator
gen = PythonCodeGenerator()
if not scalar_arg_dtypes:
@@ -189,8 +187,6 @@ def generate_specific_arg_handling_body(function_name,
# {{{ error handler
def wrap_in_error_handler(body, arg_names):
- from pytools.py_codegen import PythonCodeGenerator, Indentation
-
err_gen = PythonCodeGenerator()
def gen_error_handler():
@@ -233,8 +229,6 @@ def _generate_enqueue_and_set_args_module(function_name,
work_around_arg_count_bug, warn_about_arg_count_bug,
include_debug_helpers):
- from pytools.py_codegen import PythonCodeGenerator, Indentation
-
arg_names = ["arg%d" % i for i in range(num_passed_args)]
if scalar_arg_dtypes is None:
--
GitLab