diff options
Diffstat (limited to 'numpy')
| -rw-r--r-- | numpy/core/src/multiarray/na_mask.c | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/numpy/core/src/multiarray/na_mask.c b/numpy/core/src/multiarray/na_mask.c index 07aedfe9e..a57d530ab 100644 --- a/numpy/core/src/multiarray/na_mask.c +++ b/numpy/core/src/multiarray/na_mask.c @@ -413,13 +413,19 @@ PyArray_IsNA(PyObject *obj) return NULL; } + /* + * TODO: Could change this to use raw iteration to + * avoid the iterator creation overhead. + */ if (PyArray_HASMASKNA((PyArrayObject *)obj)) { NpyIter *iter; PyArrayObject *op[2] = {(PyArrayObject *)obj, NULL}; npy_uint32 flags, op_flags[2]; PyArray_Descr *op_dtypes[2] = {NULL, dtype}; - flags = NPY_ITER_EXTERNAL_LOOP | NPY_ITER_ZEROSIZE_OK; + flags = NPY_ITER_EXTERNAL_LOOP | + NPY_ITER_ZEROSIZE_OK | + NPY_ITER_REFS_OK; /* * This USE_MASKNA causes there to be 3 operands, where operand * 2 is the mask for operand 0 |