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

Fix GpuJoin exception and segfault.

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