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

Fix segfault in batchnorm.

上级 8757c2dc
...@@ -45,7 +45,7 @@ int dnn_batchnorm_op(PyGpuArrayObject *inp, PyGpuArrayObject *scale, ...@@ -45,7 +45,7 @@ int dnn_batchnorm_op(PyGpuArrayObject *inp, PyGpuArrayObject *scale,
#ifdef RUNNING_AVERAGES #ifdef RUNNING_AVERAGES
#ifdef INPLACE_RUNNING_MEAN #ifdef INPLACE_RUNNING_MEAN
Py_XDECREF(out_running_mean); Py_XDECREF(*out_running_mean);
PyGpuArrayObject *running_mean = in_running_mean; PyGpuArrayObject *running_mean = in_running_mean;
Py_INCREF(running_mean); Py_INCREF(running_mean);
#else #else
...@@ -56,7 +56,7 @@ int dnn_batchnorm_op(PyGpuArrayObject *inp, PyGpuArrayObject *scale, ...@@ -56,7 +56,7 @@ int dnn_batchnorm_op(PyGpuArrayObject *inp, PyGpuArrayObject *scale,
} }
#endif #endif
#ifdef INPLACE_RUNNING_VAR #ifdef INPLACE_RUNNING_VAR
Py_XDECREF(out_running_var); Py_XDECREF(*out_running_var);
PyGpuArrayObject *running_var = in_running_var; PyGpuArrayObject *running_var = in_running_var;
Py_INCREF(running_var); Py_INCREF(running_var);
#else #else
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论