binaryen.py/examples/fib.py at main · jonathanharg/binaryen.py · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
import binaryen
from binaryen.type import Int32


# Equivalent python function
def fib(n):
    if n <= 1:
        return n
    else:
        return fib(n - 1) + fib(n - 2)


mod = binaryen.Module()

n = mod.local_get(0, Int32)

condition = mod.binary(
    binaryen.operations.LeSInt32(),
    n,
    mod.i32(1),
)

n_minus_one = mod.binary(
    binaryen.operations.SubInt32(),
    n,
    mod.i32(1),
)

n_minus_two = mod.binary(
    binaryen.operations.SubInt32(),
    n,
    mod.i32(2),
)

mod.add_function(
    b"fib",
    Int32,
    Int32,
    [],
    mod.If(
        condition,
        mod.Return(n),
        mod.Return(
            mod.binary(
                binaryen.operations.AddInt32(),
                mod.call(
                    b"fib",
                    [n_minus_one],
                    Int32,
                ),
                mod.call(
                    b"fib",
                    [n_minus_two],
                    Int32,
                ),
            )
        ),
    ),
)

if not mod.validate():
    raise RuntimeError("Invalid module!")

mod.add_function_export(b"fib", b"fib")

mod.optimize()

mod.print()

# Run the written binary with `wasmtime --invoke fib fib.wasm 23`