提交 77a0eba2 authored 作者: Frederic's avatar Frederic

decref on t as this is clearner.

上级 0cef4bd1
...@@ -80,7 +80,8 @@ class CumsumOp(theano.Op): ...@@ -80,7 +80,8 @@ class CumsumOp(theano.Op):
if (!t){ if (!t){
%(fail)s; %(fail)s;
} }
Py_XDECREF(%(z)s); // Because PyArray_CumSum returns a newly created reference on %(z)s. // Because PyArray_CumSum returns a newly created reference on %(t)s.
Py_XDECREF(%(t)s);
} }
""" % locals() """ % locals()
else: else:
...@@ -101,14 +102,15 @@ class CumsumOp(theano.Op): ...@@ -101,14 +102,15 @@ class CumsumOp(theano.Op):
if (!t){ if (!t){
%(fail)s; %(fail)s;
} }
Py_XDECREF(%(z)s); // Because PyArray_CumSum returns a newly created reference on %(z)s. // Because PyArray_CumSum returns a newly created reference on %(t)s.
Py_XDECREF(%(t)s);
} }
""" % locals() """ % locals()
return code return code
def c_code_cache_version(self): def c_code_cache_version(self):
return (5,) return (6,)
def __str__(self): def __str__(self):
return "%s{%s}" % (self.__class__.__name__, self.axis) return "%s{%s}" % (self.__class__.__name__, self.axis)
...@@ -200,7 +202,8 @@ class CumprodOp(theano.Op): ...@@ -200,7 +202,8 @@ class CumprodOp(theano.Op):
if (!t){ if (!t){
%(fail)s; %(fail)s;
} }
Py_XDECREF(%(z)s); // Because PyArray_CumSum returns a newly created reference on %(z)s. // Because PyArray_CumSum returns a newly created reference on %(t)s.
Py_XDECREF(%(t)s);
} }
""" % locals() """ % locals()
else: else:
...@@ -220,14 +223,15 @@ class CumprodOp(theano.Op): ...@@ -220,14 +223,15 @@ class CumprodOp(theano.Op):
if (!t){ if (!t){
%(fail)s; %(fail)s;
} }
Py_XDECREF(%(z)s); // Because PyArray_CumSum returns a newly created reference on %(z)s. // Because PyArray_CumSum returns a newly created reference on %(t)s.
Py_XDECREF(%(t)s);
} }
""" % locals() """ % locals()
return code return code
def c_code_cache_version(self): def c_code_cache_version(self):
return (3,) return (4,)
def __str__(self): def __str__(self):
return "%s{%s}" % (self.__class__.__name__, self.axis) return "%s{%s}" % (self.__class__.__name__, self.axis)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论