提交 1dbd3e96 authored 作者: Frederic's avatar Frederic

Fix GpuJoin exception and segfault.

上级 68ede78a
......@@ -3030,7 +3030,6 @@ class GpuJoin(tensor.Join, GpuOp):
for i, cdna in enumerate(inputs[1:]):
str += """
sum += shape_%(cdna)s[axis];
Py_XDECREF(stop);
stop = PyInt_FromLong(sum);
slice_tuple = PyTuple_New(nd);
section_slice = PySlice_New(start, stop, step);
......@@ -3063,6 +3062,7 @@ class GpuJoin(tensor.Join, GpuOp):
Py_XDECREF(slice_tuple);
Py_XDECREF(start);
start = stop;
stop = NULL;
""" % locals()
str+="""
......@@ -3072,7 +3072,7 @@ class GpuJoin(tensor.Join, GpuOp):
return str
def c_code_cache_version(self):
return (1,)
return (2,)
gpu_join = GpuJoin()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论