提交 c9b28cc2 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Address comments.

上级 8dc7320f
...@@ -221,6 +221,7 @@ class GpuElemwise(HideC, Elemwise): ...@@ -221,6 +221,7 @@ class GpuElemwise(HideC, Elemwise):
# check that all inputs have valid dimensions # check that all inputs have valid dimensions
emitted_inames = {} emitted_inames = {}
code = """ code = """
// +1 is so that MSVC is happy when nd == 0
size_t dims[%(nd)s+1] = {%(initial_dims)s}; size_t dims[%(nd)s+1] = {%(initial_dims)s};
void *rargs[%(nargs)s] = {0}; void *rargs[%(nargs)s] = {0};
""" % locals() """ % locals()
...@@ -269,7 +270,7 @@ class GpuElemwise(HideC, Elemwise): ...@@ -269,7 +270,7 @@ class GpuElemwise(HideC, Elemwise):
%(fail)s; %(fail)s;
} }
} }
""" % locals() """ % locals()
emitted_inames[iname] = True emitted_inames[iname] = True
# check that all outputs have valid dimensions # check that all outputs have valid dimensions
p = len(node.inputs) p = len(node.inputs)
...@@ -299,7 +300,7 @@ class GpuElemwise(HideC, Elemwise): ...@@ -299,7 +300,7 @@ class GpuElemwise(HideC, Elemwise):
} }
} }
rargs[%(p)s] = &%(oname)s->ga; rargs[%(p)s] = &%(oname)s->ga;
""" % locals() """ % locals()
p += 1 p += 1
else: else:
input_idx = self.inplace_pattern[idx] input_idx = self.inplace_pattern[idx]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论