Commits

Maciej Fijalkowski committed 2644eca

fix the merge point in flat_set_driver

Comments (0)

Files changed (1)

pypy/module/micronumpy/interp_numarray.py

 )
 flat_set_driver = jit.JitDriver(
     greens=['shapelen', 'base'],
-    reds=['step', 'ai', 'lngth', 'arr', 'basei'],
+    reds=['step', 'ai', 'ri', 'lngth', 'arr', 'basei'],
     name='numpy_flatset',
 )
 
                                              basei=basei,
                                              step=step,
                                              res=res,
-                                             ri=ri,
-                                            )
+                                             ri=ri)
             w_val = base.getitem(basei.offset)
             res.setitem(ri.offset, w_val)
             basei = basei.next_skip_x(shapelen, step)
         basei = basei.next_skip_x(shapelen, start)
         while lngth > 0:
             flat_set_driver.jit_merge_point(shapelen=shapelen,
-                                             basei=basei,
-                                             base=base,
-                                             step=step,
-                                             arr=arr,
-                                             ai=ai,
-                                             lngth=lngth,
-                                            )
+                                            basei=basei,
+                                            base=base,
+                                            step=step,
+                                            arr=arr,
+                                            ai=ai,
+                                            lngth=lngth,
+                                            ri=ri)
             v = arr.getitem(ri.offset).convert_to(base.dtype)
             base.setitem(basei.offset, v)
             # need to repeat input values until all assignments are done