提交 c575da2c authored 作者: Brandon T. Willard's avatar Brandon T. Willard 提交者: Brandon T. Willard

Fix bad Windows free in dimshuffle.c

上级 c34f37f1
...@@ -37,11 +37,10 @@ int APPLY_SPECIFIC(cpu_dimshuffle)(PyArrayObject *input, PyArrayObject **res, ...@@ -37,11 +37,10 @@ int APPLY_SPECIFIC(cpu_dimshuffle)(PyArrayObject *input, PyArrayObject **res,
npy_intp N_shuffle = PyArray_SIZE(params->shuffle); npy_intp N_shuffle = PyArray_SIZE(params->shuffle);
npy_intp N_augment = PyArray_SIZE(params->augment); npy_intp N_augment = PyArray_SIZE(params->augment);
npy_intp N = N_augment + N_shuffle; npy_intp N = N_augment + N_shuffle;
npy_intp *_reshape_shape = (npy_intp *)malloc(N * sizeof(npy_intp)); npy_intp *_reshape_shape = PyDimMem_NEW(N);
if (_reshape_shape == NULL) { if (_reshape_shape == NULL) {
PyErr_NoMemory(); PyErr_NoMemory();
free(_reshape_shape);
return 1; return 1;
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论