提交 76ba144d authored 作者: sebastien's avatar sebastien

Make changes according to Arnaud's comments

上级 50523ea7
...@@ -184,7 +184,6 @@ inplace_increment(PyObject *dummy, PyObject *args) ...@@ -184,7 +184,6 @@ inplace_increment(PyObject *dummy, PyObject *args)
return NULL; return NULL;
} }
mit = (PyArrayMapIterObject *) PyArray_MapIterArray(a, index); mit = (PyArrayMapIterObject *) PyArray_MapIterArray(a, index);
Py_INCREF(mit);
if (mit == NULL) { if (mit == NULL) {
goto fail; goto fail;
} }
......
...@@ -1943,7 +1943,6 @@ class AdvancedIncSubtensor1(Op): ...@@ -1943,7 +1943,6 @@ class AdvancedIncSubtensor1(Op):
return NULL; return NULL;
} }
mit = (PyArrayMapIterObject *) PyArray_MapIterArray(a, index); mit = (PyArrayMapIterObject *) PyArray_MapIterArray(a, index);
Py_INCREF(mit); // Should I INCREF here?
if (mit == NULL) { if (mit == NULL) {
goto fail; goto fail;
} }
...@@ -1962,18 +1961,7 @@ class AdvancedIncSubtensor1(Op): ...@@ -1962,18 +1961,7 @@ class AdvancedIncSubtensor1(Op):
return NULL; return NULL;
} }
#endif #endif
""")
PyMethodDef CutilsExtMethods[] = {
{"run_cthunk", run_cthunk, METH_VARARGS|METH_KEYWORDS,
"Run a theano cthunk."},
#if NPY_API_VERSION >= 0x00000008
{"inplace_increment", inplace_increment,
METH_VARARGS,
"increments a numpy array inplace at the passed indexes."},
#endif
{NULL, NULL, 0, NULL} /* Sentinel */
};""")
def c_code(self, node, name, input_names, output_names, sub): def c_code(self, node, name, input_names, output_names, sub):
x, y, idx = input_names x, y, idx = input_names
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论