[pypy-commit] pypy ufuncapi: translation fix

[mattip]

Author: mattip <matti.picus at gmail.com>
Branch: ufuncapi
Changeset: r74829:9d7d30ef07d4
Date: 2014-12-05 12:53 +0200
http://bitbucket.org/pypy/pypy/changeset/9d7d30ef07d4/

Log:	translation fix

diff --git a/pypy/module/micronumpy/ufuncs.py b/pypy/module/micronumpy/ufuncs.py
--- a/pypy/module/micronumpy/ufuncs.py
+++ b/pypy/module/micronumpy/ufuncs.py
@@ -825,16 +825,17 @@
             dim_offset = self.core_offsets[i]
             num_dims = self.core_num_dims[i]
             if not isinstance(curarg, W_NDimArray):
-                arg_shape = iter_shape[:]
+                target_dims = []
                 for j in range(num_dims):
                     core_dim_index = self.core_dim_ixs[dim_offset + j]
-                    if matched_dims[core_dim_index] < 0:
+                    v = matched_dims[core_dim_index]
+                    if v < 0:
                         raise oefmt(space.w_ValueError, "%s: %s operand %d "
                             "is empty but unique core dimension %d in signature "
                             "%s of gufunc was not specified",
                              self.name, name, _i, core_dim_index, self.signature)
-                    arg_shape.append(matched_dims[core_dim_index])
-                arg_shapes.append(arg_shape) 
+                    target_dims.append(v)
+                arg_shapes.append(iter_shape + target_dims) 
                 continue
             n = len(curarg.get_shape()) - num_dims
             if n < 0: