提交 4e27ad46 authored 作者: Frederic's avatar Frederic

Use current numpy c-api

上级 4b4eeaad
...@@ -155,11 +155,11 @@ class WeirdBrokenOp(gof.Op): ...@@ -155,11 +155,11 @@ class WeirdBrokenOp(gof.Op):
prep_vars = """ prep_vars = """
//the output array has size M x N //the output array has size M x N
npy_intp M = PyArray_DIMS(%(a)s)[0]; npy_intp M = PyArray_DIMS(%(a)s)[0];
npy_intp Sa = %(a)s->strides[0] / PyArray_DESCR(%(a)s)->elsize; npy_intp Sa = PyArray_STRIDES(%(a)s)[0] / PyArray_DESCR(%(a)s)->elsize;
npy_intp Sz = %(z)s->strides[0] / PyArray_DESCR(%(z)s)->elsize; npy_intp Sz = PyArray_STRIDES(%(z)s)[0] / PyArray_DESCR(%(z)s)->elsize;
npy_double * Da = (npy_double*)%(a)s->data; npy_double * Da = (npy_double*)PyArray_BYTES(%(a)s);
npy_double * Dz = (npy_double*)%(z)s->data; npy_double * Dz = (npy_double*)PyArray_BYTES(%(z)s);
//clear the output array //clear the output array
for (npy_intp m = 0; m < M; ++m) for (npy_intp m = 0; m < M; ++m)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论