提交 8d26859e authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Allow string constants in kernel code to actually work.

上级 cff861bf
...@@ -119,6 +119,8 @@ class GpuKernelBase(object): ...@@ -119,6 +119,8 @@ class GpuKernelBase(object):
vname = self.c_kernel_codevar(name) vname = self.c_kernel_codevar(name)
kname = self.c_kernel_obj(name) kname = self.c_kernel_obj(name)
code = '\\n'.join(l for l in kcode.split('\n')) code = '\\n'.join(l for l in kcode.split('\n'))
code = code.replace('"', '\\"')
print code
return """static const char *%(vname)s = "%(code)s"; return """static const char *%(vname)s = "%(code)s";
static GpuKernel %(kname)s;""" % dict(vname=vname, kname=kname, code=code) static GpuKernel %(kname)s;""" % dict(vname=vname, kname=kname, code=code)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论