[pypy-commit] pypy nditer-external_loop: start implementing nditer external_loop
mattip
noreply at buildbot.pypy.org
Wed Oct 29 23:09:03 CET 2014
Author: mattip <matti.picus at gmail.com>
Branch: nditer-external_loop
Changeset: r74286:b7140dce25c7
Date: 2014-10-21 18:00 +0200
http://bitbucket.org/pypy/pypy/changeset/b7140dce25c7/
Log: start implementing nditer external_loop
diff --git a/pypy/module/micronumpy/nditer.py b/pypy/module/micronumpy/nditer.py
--- a/pypy/module/micronumpy/nditer.py
+++ b/pypy/module/micronumpy/nditer.py
@@ -186,13 +186,14 @@
ndim = max(ndim, op.ndims())
return ndim
-def coalexce_axes(iter, space):
+def coalexce_axes(it, space):
# Copy logic from npyiter_coalesce_axes, used in ufunc iterators
# and in nditer's with 'external_loop' flag
- import pdb;pdb.set_trace()
- for idim in range(iter.ndim - 1):
+ for idim in range(it.ndim - 1):
can_coalesce = 1
- for op in self.ops:
+ for op in it.seq:
+ stride = op.implementation.get_strides()
+ shape = op.get_shape()
pass
class IndexIterator(object):
More information about the pypy-commit
mailing list