Esempio n. 1
0
 def compute(self):
     result = W_NDimArray(self.size, self.shape, self.find_dtype())
     shapelen = len(self.shape)
     sig = self.find_sig()
     frame = sig.create_frame(self)
     ri = ArrayIterator(self.size)
     while not ri.done():
         numpy_driver.jit_merge_point(sig=sig,
                                      shapelen=shapelen,
                                      result_size=self.size,
                                      frame=frame,
                                      ri=ri,
                                      self=self, result=result)
         result.dtype.setitem(result.storage, ri.offset,
                              sig.eval(frame, self))
         frame.next(shapelen)
         ri = ri.next(shapelen)
     return result
Esempio n. 2
0
 def compute(self):
     result = W_NDimArray(self.size, self.shape, self.find_dtype())
     shapelen = len(self.shape)
     sig = self.find_sig()
     frame = sig.create_frame(self)
     ri = ArrayIterator(self.size)
     while not ri.done():
         numpy_driver.jit_merge_point(sig=sig,
                                      shapelen=shapelen,
                                      result_size=self.size,
                                      frame=frame,
                                      ri=ri,
                                      self=self,
                                      result=result)
         result.dtype.setitem(result.storage, ri.offset,
                              sig.eval(frame, self))
         frame.next(shapelen)
         ri = ri.next(shapelen)
     return result