Reduced allocations in run

Author: GTof

Makefile

@@ -29,3 +29,6 @@ docs:
 
 opt: clean
 	./opt.sh
+
+profile: opt
+	cd opt && python -m cProfile ../benchmarks/imeprofile.py 30 | less

benchmarks/fibonacci.py

@@ -8,28 +8,26 @@
 t = int(sys.argv[2])
 
 
-def fibo(u0: int, u1: int, i: int) -> int:
+def fibo(i: int) -> int:
     if i > 1:
-        return fibo(u1, u0 + u1, i - 1)
-    if i == 1:
-        return u1
-    return u0
+        return fibo(i - 1) + fibo(i - 2)
+    else:
+        return i
 
 
-def fibo_io(u0: int, u1: int, i: int) -> IO:
+def fibo_io(i: int) -> IO:
     if i > 1:
-        return io.defer_io(lambda: fibo_io(u1, u0 + u1, i - 1))
-    if i == 1:
-        return io.pure(u1)
-    return io.pure(u0)
+        return io.defer_io(lambda: fibo_io(i-1)).flat_map(lambda x: fibo_io(i-2).map(lambda y: x+y))
+    else:
+        return io.pure(i)
 
 
 def mesure(nb):
     l = []
     i = 0
-    while i < 10 * nb:
+    while i < nb:
         start = timer()
-        fibo(0, 1, n)
+        fibo(n)
         end = timer()
         l.append(end - start)
         i += 1
@@ -39,7 +37,7 @@ def mesure(nb):
 def mesure_io(nb):
     l = []
     i = 0
-    mio = fibo_io(0, 1, n)
+    mio = fibo_io(n)
     while i < nb:
         start = timer()
         mio.run(None)
@@ -48,6 +46,10 @@ def mesure_io(nb):
         i += 1
     return statistics.median(l)
 
+x = 8
+
+print(f"fibo({x})=${fibo(x)}")
+print(f"fibo_io({x})=${fibo_io(x).run(None).raise_on_panic()}")
 
 print(f"fibo({n})    : {mesure(t)}")
 print(f"fibo_io({n}) : {mesure_io(t)}")

benchmarks/imeprofile.py

@@ -4,15 +4,14 @@
 import sys
 
 
-def fibo_io(u0: int, u1: int, i: int) -> IO:
+def fibo_io(i: int) -> IO:
     if i > 1:
-        return io.defer_io(lambda: fibo_io(u1, (u0 + u1) % 10000, i - 1))
-    if i == 1:
-        return io.pure(u1)
-    return io.pure(u0)
+        return io.defer_io(lambda: fibo_io(i - 1)).flat_map(lambda x: fibo_io(i-2).map(lambda y: x + y))
+    else:
+        return io.pure(i)
 
 
-m = fibo_io(0, 1, int(sys.argv[1]))
+m = fibo_io(int(sys.argv[1]))
 
 
 profiler = cProfile.Profile()

docs/io.html

@@ -67,6 +67,8 @@ <h1 class="title">Module <code>raffiot.io</code></h1>
     Have a look to the documentation and examples to learn how to use it.
     &#34;&#34;&#34;
 
+    __slots__ = &#34;__tag&#34;, &#34;__fields&#34;
+
     @final
     def __init__(self, __tag, __fields):
         self.__tag = __tag
@@ -78,8 +80,6 @@ <h1 class="title">Module <code>raffiot.io</code></h1>
         Transform the computed value with f if the computation is successful.
         Do nothing otherwise.
         &#34;&#34;&#34;
-        if self.__tag in [9, 12]:
-            return self
         return IO(1, (self, f))
 
     @final
@@ -107,8 +107,6 @@ <h1 class="title">Module <code>raffiot.io</code></h1>
         and arg computes a value `x: X`
         then self.ap(arg) computes `f(x): A`
         &#34;&#34;&#34;
-        if self.__tag == 0 and arg.__tag == 0:
-            return IO(5, lambda: self.__fields(arg.__fields))
         return IO(3, (self, arg))
 
     @final
@@ -128,8 +126,6 @@ <h1 class="title">Module <code>raffiot.io</code></h1>
         Transform the context with f.
         Note that f is not from R to R2 but from R2 to R!
         &#34;&#34;&#34;
-        if self.__tag in [0, 5, 9, 12]:
-            return self
         return IO(8, (f, self))
 
     # Error API
@@ -141,8 +137,6 @@ <h1 class="title">Module <code>raffiot.io</code></h1>
 
         On error, call the handler with the error.
         &#34;&#34;&#34;
-        if self.__tag in [0, 5, 7, 12]:
-            return self
         return IO(10, (self, handler))
 
     @final
@@ -151,8 +145,6 @@ <h1 class="title">Module <code>raffiot.io</code></h1>
         Transform the stored error if the computation fails on an error.
         Do nothing otherwise.
         &#34;&#34;&#34;
-        if self.__tag in [0, 5, 7, 12]:
-            return self
         return IO(11, (self, f))
 
     # Panic
@@ -164,8 +156,6 @@ <h1 class="title">Module <code>raffiot.io</code></h1>
 
         On panic, call the handler with the exception.
         &#34;&#34;&#34;
-        if self.__tag in [0, 7, 9]:
-            return self
         return IO(13, (self, handler))
 
     @final
@@ -174,8 +164,6 @@ <h1 class="title">Module <code>raffiot.io</code></h1>
         Transform the exception stored if the computation fails on a panic.
         Do nothing otherwise.
         &#34;&#34;&#34;
-        if self.__tag in [0, 7, 9]:
-            return self
         return IO(14, (self, f))
 
     @final
@@ -841,6 +829,8 @@ <h2 class="section-title" id="header-classes">Classes</h2>
     Have a look to the documentation and examples to learn how to use it.
     &#34;&#34;&#34;
 
+    __slots__ = &#34;__tag&#34;, &#34;__fields&#34;
+
     @final
     def __init__(self, __tag, __fields):
         self.__tag = __tag
@@ -852,8 +842,6 @@ <h2 class="section-title" id="header-classes">Classes</h2>
         Transform the computed value with f if the computation is successful.
         Do nothing otherwise.
         &#34;&#34;&#34;
-        if self.__tag in [9, 12]:
-            return self
         return IO(1, (self, f))
 
     @final
@@ -881,8 +869,6 @@ <h2 class="section-title" id="header-classes">Classes</h2>
         and arg computes a value `x: X`
         then self.ap(arg) computes `f(x): A`
         &#34;&#34;&#34;
-        if self.__tag == 0 and arg.__tag == 0:
-            return IO(5, lambda: self.__fields(arg.__fields))
         return IO(3, (self, arg))
 
     @final
@@ -902,8 +888,6 @@ <h2 class="section-title" id="header-classes">Classes</h2>
         Transform the context with f.
         Note that f is not from R to R2 but from R2 to R!
         &#34;&#34;&#34;
-        if self.__tag in [0, 5, 9, 12]:
-            return self
         return IO(8, (f, self))
 
     # Error API
@@ -915,8 +899,6 @@ <h2 class="section-title" id="header-classes">Classes</h2>
 
         On error, call the handler with the error.
         &#34;&#34;&#34;
-        if self.__tag in [0, 5, 7, 12]:
-            return self
         return IO(10, (self, handler))
 
     @final
@@ -925,8 +907,6 @@ <h2 class="section-title" id="header-classes">Classes</h2>
         Transform the stored error if the computation fails on an error.
         Do nothing otherwise.
         &#34;&#34;&#34;
-        if self.__tag in [0, 5, 7, 12]:
-            return self
         return IO(11, (self, f))
 
     # Panic
@@ -938,8 +918,6 @@ <h2 class="section-title" id="header-classes">Classes</h2>
 
         On panic, call the handler with the exception.
         &#34;&#34;&#34;
-        if self.__tag in [0, 7, 9]:
-            return self
         return IO(13, (self, handler))
 
     @final
@@ -948,8 +926,6 @@ <h2 class="section-title" id="header-classes">Classes</h2>
         Transform the exception stored if the computation fails on a panic.
         Do nothing otherwise.
         &#34;&#34;&#34;
-        if self.__tag in [0, 7, 9]:
-            return self
         return IO(14, (self, f))
 
     @final
@@ -1219,8 +1195,6 @@ <h3>Methods</h3>
     and arg computes a value `x: X`
     then self.ap(arg) computes `f(x): A`
     &#34;&#34;&#34;
-    if self.__tag == 0 and arg.__tag == 0:
-        return IO(5, lambda: self.__fields(arg.__fields))
     return IO(3, (self, arg))</code></pre>
 </details>
 </dd>
@@ -1284,8 +1258,6 @@ <h3>Methods</h3>
 
     On error, call the handler with the error.
     &#34;&#34;&#34;
-    if self.__tag in [0, 5, 7, 12]:
-        return self
     return IO(10, (self, handler))</code></pre>
 </details>
 </dd>
@@ -1305,8 +1277,6 @@ <h3>Methods</h3>
     Transform the context with f.
     Note that f is not from R to R2 but from R2 to R!
     &#34;&#34;&#34;
-    if self.__tag in [0, 5, 9, 12]:
-        return self
     return IO(8, (f, self))</code></pre>
 </details>
 </dd>
@@ -1390,8 +1360,6 @@ <h3>Methods</h3>
     Transform the computed value with f if the computation is successful.
     Do nothing otherwise.
     &#34;&#34;&#34;
-    if self.__tag in [9, 12]:
-        return self
     return IO(1, (self, f))</code></pre>
 </details>
 </dd>
@@ -1411,8 +1379,6 @@ <h3>Methods</h3>
     Transform the stored error if the computation fails on an error.
     Do nothing otherwise.
     &#34;&#34;&#34;
-    if self.__tag in [0, 5, 7, 12]:
-        return self
     return IO(11, (self, f))</code></pre>
 </details>
 </dd>
@@ -1432,8 +1398,6 @@ <h3>Methods</h3>
     Transform the exception stored if the computation fails on a panic.
     Do nothing otherwise.
     &#34;&#34;&#34;
-    if self.__tag in [0, 7, 9]:
-        return self
     return IO(14, (self, f))</code></pre>
 </details>
 </dd>
@@ -1492,8 +1456,6 @@ <h3>Methods</h3>
 
     On panic, call the handler with the exception.
     &#34;&#34;&#34;
-    if self.__tag in [0, 7, 9]:
-        return self
     return IO(13, (self, handler))</code></pre>
 </details>
 </dd>