optimised benchmark

4f65182c · Stefan Behnel · 3e37c134 · 4f65182c
Commit 4f65182c authored Mar 31, 2012 by Stefan Behnel
Hide whitespace changes
Inline Side-by-side

Showing with 5 additions and 2 deletions

Demos/benchmarks/generators.py Demos/benchmarks/generators.py +5 -2

No files found.
--- a/Demos/benchmarks/generators.py
+++ b/Demos/benchmarks/generators.py
@@ -3,12 +3,15 @@

 COUNT = 10000

+import cython
+
+@cython.locals(N=cython.Py_ssize_t)
 def count_to(N):
    for i in range(N):
        yield i

-def round_robin(*iterators):
-    iterators = list(iterators)
+def round_robin(*_iterators):
+    iterators = list(_iterators)
    to_drop = []
    while iterators:
        for i, it in enumerate(iterators):