提交 db36ac5d authored 作者: affanv14's avatar affanv14

change macro names

上级 56da8ca8
...@@ -499,7 +499,7 @@ class CGpuKernelBase(COp, GpuKernelBase): ...@@ -499,7 +499,7 @@ class CGpuKernelBase(COp, GpuKernelBase):
undef_macros = [] undef_macros = []
for i, v in enumerate(node.inputs): for i, v in enumerate(node.inputs):
if isinstance(v.type, GpuArrayType): if isinstance(v.type, GpuArrayType):
macro_name = "DTYPE_i%d" % (i,) macro_name = "DTYPE_INPUT_%d" % (i,)
macro_value = pygpu.gpuarray.dtype_to_ctype(v.dtype) macro_value = pygpu.gpuarray.dtype_to_ctype(v.dtype)
define_macros.append( define_macros.append(
define_template % define_template %
...@@ -507,7 +507,7 @@ class CGpuKernelBase(COp, GpuKernelBase): ...@@ -507,7 +507,7 @@ class CGpuKernelBase(COp, GpuKernelBase):
undef_macros.append(undef_template % macro_name) undef_macros.append(undef_template % macro_name)
for i, v in enumerate(node.outputs): for i, v in enumerate(node.outputs):
if isinstance(v.type, GpuArrayType): if isinstance(v.type, GpuArrayType):
macro_name = "DTYPE_o%d" % (i,) macro_name = "DTYPE_OUTPUT_%d" % (i,)
macro_value = pygpu.gpuarray.dtype_to_ctype(v.dtype) macro_value = pygpu.gpuarray.dtype_to_ctype(v.dtype)
define_macros.append( define_macros.append(
define_template % define_template %
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论