Skip to content

Commit 711af43

Browse files
committed
introduced a jit for reducing vectors
1 parent 08348f3 commit 711af43

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

pixie/vm/persistent_vector.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -176,12 +176,20 @@ def _push(self, v):
176176
def _push(self):
177177
return self.pop()
178178

179+
180+
_reduce_driver = jit.JitDriver(name="pixie.stdlib.PersistentVector_reduce",
181+
greens=["f"],
182+
reds="auto")
183+
179184
@extend(proto._reduce, PersistentVector)
180185
def _reduce(self, f, init):
181186
i = 0
182187
while i < self._cnt:
183188
array = self.array_for(i)
184189
for j in range(len(array)):
190+
item = array[j]
191+
_reduce_driver.jit_merge_point(f=f)
192+
185193
init = f.invoke([init, array[j]])
186194
if rt.reduced_QMARK_(init):
187195
return rt.deref(init)

0 commit comments

Comments
 (0)