diff --git a/loopy/target/pyopencl.py b/loopy/target/pyopencl.py index 8f371085e0f1655651397c16873f10a95a799f79..f24b115fd5a35af94e4a6d437550bccf86b5bee0 100644 --- a/loopy/target/pyopencl.py +++ b/loopy/target/pyopencl.py @@ -335,6 +335,8 @@ class PyOpenCLTarget(OpenCLTarget): % dev_id) def preprocess(self, kernel): + if self.device is not None: + kernel = adjust_local_temp_var_storage(kernel, self.device) return kernel def pre_codegen_check(self, kernel):