224fdc7a78
Motivation: in 64-bit machines, we can store boxed uint32 values as a tagged pointer. In 32-bit machines, we need to allocated an object (like Haskell) to store the uint32 value. So, the generated bytecode is quite different in each platform. This change also allow us to simplify the IR. Example: we don't need the type `sizet` anymore. Impact: To be able to bootstrap in both platforms, we will have to store two versions of the generated code: 32 and 64 versions. In principle, we only need to store the 64-bit version, and use cross-compilation to build the 32-bit version.
90 lines
2.1 KiB
Text
90 lines
2.1 KiB
Text
import system.io
|
|
import init.lean.ir.parser init.lean.ir.format
|
|
import init.lean.ir.elim_phi init.lean.ir.type_check
|
|
import init.lean.ir.extract_cpp
|
|
|
|
open lean.parser
|
|
open lean.ir
|
|
|
|
def check_decl (d : decl) : io unit :=
|
|
match type_check d with
|
|
| except.ok _ := return ()
|
|
| except.error e := io.print_ln (to_string e)
|
|
|
|
def show_result (p : parser decl) (s : string) : io unit :=
|
|
match parse p s with
|
|
| except.ok d := io.print_ln (lean.to_fmt d) >> check_decl d
|
|
| except.error e := io.print_ln (e.to_string s)
|
|
|
|
def IR1 := "
|
|
def succ (x : uint32) : uint32 :=
|
|
main: one : uint32 := 1; x1 : uint32 := add x one; ret x1;
|
|
"
|
|
|
|
#eval show_result (whitespace >> parse_def) IR1
|
|
|
|
def IR2 := "
|
|
def add_n (x : uint32) (y : uint32) (n : uint32) : uint32 :=
|
|
main: jmp loop;
|
|
loop:
|
|
r1 : uint32 := phi x r2;
|
|
y1 : uint32 := phi y y1;
|
|
n1 : uint32 := phi n n2;
|
|
r2 : uint32 := add r1 y1;
|
|
one : uint32 := 1;
|
|
n2 : uint32 := sub n1 one;
|
|
zero : uint32 := 0;
|
|
c : bool := eq n2 zero;
|
|
case c [loop, end];
|
|
end:
|
|
r3 : uint32 := phi r2;
|
|
ret r3;
|
|
"
|
|
|
|
#eval show_result (whitespace >> parse_def) IR2
|
|
|
|
def tst_elim_phi (s : string) : io unit :=
|
|
do (except.ok d) ← return $ parse (whitespace >> parse_def) s,
|
|
check_decl d,
|
|
io.print_ln (lean.to_fmt (elim_phi d))
|
|
|
|
#eval tst_elim_phi IR2
|
|
|
|
def IR3 := "
|
|
def mk_struct (d1 : object) (d2 : uint32) (d3 : usize) (d4 : uint32) (d5 : bool) (d6 : bool) : object :=
|
|
main:
|
|
o := cnstr 0 1 18;
|
|
set o 0 d1;
|
|
sset o 8 d3;
|
|
sset o 16 d2;
|
|
sset o 20 d4;
|
|
sset o 24 d5;
|
|
sset o 25 d6;
|
|
ret o;
|
|
"
|
|
#eval show_result (whitespace >> parse_def) IR3
|
|
|
|
def tst_extract_cpp (s : string) : io unit :=
|
|
do (except.ok d) ← return $ parse (whitespace >> parse_def) s,
|
|
check_decl d,
|
|
(except.ok code) ← return $ extract_cpp (λ _, none) (λ _, none) [elim_phi d],
|
|
io.print_ln code
|
|
|
|
#eval tst_extract_cpp IR3
|
|
#eval tst_extract_cpp IR2
|
|
|
|
def IR4 := "
|
|
def swap (d1 : object) (d2 : object) : object object :=
|
|
main:
|
|
r1 := cnstr 0 2 0;
|
|
r2 := cnstr 0 2 0;
|
|
set r1 0 d1;
|
|
set r1 1 d2;
|
|
inc d1;
|
|
inc d2;
|
|
set r2 0 d2;
|
|
set r2 1 d1;
|
|
ret r1 r2;
|
|
"
|
|
|
|
#eval tst_extract_cpp IR4
|