From 89291ac6dcd1e704c128b0254ab7ef2adc61a55f Mon Sep 17 00:00:00 2001
From: arghdos <arghdos@gmail.com>
Date: Thu, 4 May 2017 14:13:48 -0400
Subject: [PATCH] fix editing
---
loopy/target/c/c_execution.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/loopy/target/c/c_execution.py b/loopy/target/c/c_execution.py
index e6a1bd0d6..2a60ace79 100644
--- a/loopy/target/c/c_execution.py
+++ b/loopy/target/c/c_execution.py
@@ -109,13 +109,13 @@ class CompiledCKernel(object):
to automatically map argument types.
"""
- def __init__(self, knl, target, comp=None):
+ def __init__(self, knl, dev_code, target, comp=None):
from loopy.target.c import CTarget
assert isinstance(target, CTarget)
self.target = target
self.knl = knl
# get code and build
- self.code = str(knl.ast)
+ self.code = dev_code
self.comp = comp or CCompiler()
self.dll = self.comp.build(self.code)
# get the function declaration for interface with ctypes
@@ -235,11 +235,12 @@ class CKernelExecutor(KernelExecutorBase):
if self.kernel.options.edit_cl:
from pytools import invoke_editor
- dev_code = invoke_editor(dev_code, "code.cl")
+ dev_code = invoke_editor(dev_code, "code.c")
c_kernels = []
for dp in codegen_result.device_programs:
- c_kernels.append(CompiledCKernel(dp, self.kernel.target, self.compiler))
+ c_kernels.append(CompiledCKernel(dp, dev_code, self.kernel.target,
+ self.compiler))
return _KernelInfo(
kernel=kernel,
--
GitLab