From 1e7ff424255137d0b84ff718a0a98d17880c8633 Mon Sep 17 00:00:00 2001
From: arghdos <arghdos@gmail.com>
Date: Tue, 2 May 2017 13:57:34 -0400
Subject: [PATCH] add missing pyopencl fetches
---
loopy/target/pyopencl_execution.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/loopy/target/pyopencl_execution.py b/loopy/target/pyopencl_execution.py
index e67c49c59..b6c58ce47 100644
--- a/loopy/target/pyopencl_execution.py
+++ b/loopy/target/pyopencl_execution.py
@@ -180,6 +180,10 @@ class PyOpenCLExecutionWrapperGenerator(ExecutionWrapperGeneratorBase):
if not issubclass(arg.arg_class, KernelArgument):
continue
+ is_written = arg.base_name in kernel.get_written_variables()
+ if is_written:
+ gen("%s = %s.get(queue=queue)" % (arg.name, arg.name))
+
gen("")
if options.return_dict:
--
GitLab