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

Fix compilation crash

上级 5429c30a
...@@ -403,6 +403,7 @@ class GpuFromHost(Op): ...@@ -403,6 +403,7 @@ class GpuFromHost(Op):
return """ return """
PyArrayObject *%(name)s_tmp; PyArrayObject *%(name)s_tmp;
%(name)s_tmp = PyArray_GETCONTIGUOUS(%(inp)s); %(name)s_tmp = PyArray_GETCONTIGUOUS(%(inp)s);
int err;
if (%(name)s_tmp == NULL) if (%(name)s_tmp == NULL)
%(fail)s %(fail)s
...@@ -411,8 +412,8 @@ class GpuFromHost(Op): ...@@ -411,8 +412,8 @@ class GpuFromHost(Op):
(size_t *)PyArray_DIMS(%(name)s_tmp), (size_t *)PyArray_DIMS(%(name)s_tmp),
get_typecode((PyObject *)PyArray_DESCR(%(name)s_tmp)))) { get_typecode((PyObject *)PyArray_DESCR(%(name)s_tmp)))) {
Py_BEGIN_ALLOW_THREADS Py_BEGIN_ALLOW_THREADS
int err = GpuArray_write(&%(out)s->ga, PyArray_DATA(%(name)s_tmp), err = GpuArray_write(&%(out)s->ga, PyArray_DATA(%(name)s_tmp),
PyArray_NBYTES(%(name)s_tmp)); PyArray_NBYTES(%(name)s_tmp));
Py_END_ALLOW_THREADS Py_END_ALLOW_THREADS
Py_DECREF(%(name)s_tmp); Py_DECREF(%(name)s_tmp);
if (err != GA_NO_ERROR) { if (err != GA_NO_ERROR) {
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论