提交 7b09a215 authored 作者: Alexander Matyasko's avatar Alexander Matyasko

Fix flags after pygpu_narrow

上级 3b9783bc
...@@ -25,6 +25,7 @@ static PyGpuArrayObject *pygpu_narrow(PyGpuArrayObject *src, size_t dim, ...@@ -25,6 +25,7 @@ static PyGpuArrayObject *pygpu_narrow(PyGpuArrayObject *src, size_t dim,
size_t size) { size_t size) {
PyGpuArrayObject *src_view = pygpu_view(src, Py_None); PyGpuArrayObject *src_view = pygpu_view(src, Py_None);
src_view->ga.dimensions[dim] = size; src_view->ga.dimensions[dim] = size;
GpuArray_fix_flags(&src_view->ga);
return pygpu_copy(src_view, GA_C_ORDER); return pygpu_copy(src_view, GA_C_ORDER);
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论