forked from AbleOS/holey-bytes
implementing the loop iteration optimization
the multiplication and addition to a pointer is replaced with simply incremrnting the pointer it self Signed-off-by: Jakub Doka <jakub.doka2@gmail.com>
This commit is contained in:
parent
5a7a01ca02
commit
7837eeb90d
113
lang/src/son.rs
113
lang/src/son.rs
|
@ -1068,47 +1068,72 @@ impl Nodes {
|
||||||
return Some(self.new_node(ty, K::BinOp { op }, [ctrl, a, new_rhs], tys));
|
return Some(self.new_node(ty, K::BinOp { op }, [ctrl, a, new_rhs], tys));
|
||||||
}
|
}
|
||||||
|
|
||||||
//if op == T::Add
|
if op == T::Add
|
||||||
// && self[rhs].kind == (K::BinOp { op: T::Mul })
|
&& self[rhs].kind == (K::BinOp { op: T::Mul })
|
||||||
// && let &[_, mul_lhs, mul_rhs] = self[rhs].inputs.as_slice()
|
&& let &[_, index, step] = self[rhs].inputs.as_slice()
|
||||||
// && self[mul_lhs].kind == K::Phi
|
&& self[index].kind == K::Phi
|
||||||
// && let &[phi_ctrl, phi_lhs, phi_rhs] = self[mul_lhs].inputs.as_slice()
|
&& let &[iter_loop, index_init, new_index] = self[index].inputs.as_slice()
|
||||||
// && phi_rhs != VOID
|
&& new_index != VOID
|
||||||
// && self[phi_ctrl].kind == K::Loop
|
&& self[iter_loop].kind == K::Loop
|
||||||
// && self[phi_rhs].kind == (K::BinOp { op: T::Add })
|
&& self[new_index].kind == (K::BinOp { op: T::Add })
|
||||||
// && self[phi_rhs].inputs[1] == mul_lhs
|
&& self[new_index].inputs[1] == index
|
||||||
//{
|
&& let Some(&iter_cond) = self[index].outputs.iter().find(
|
||||||
// debug_assert_eq!(
|
|&&n| matches!(self[n].kind, Kind::BinOp { op } if op.is_compatison()),
|
||||||
// self[mul_lhs].outputs.len(),
|
)
|
||||||
// 1,
|
&& self[index].outputs.iter().all(|n| [iter_cond, rhs, new_index].contains(n))
|
||||||
// "{:?}",
|
{
|
||||||
// self[mul_lhs]
|
// arr := @as([u32; 10], idk)
|
||||||
// .outputs
|
//
|
||||||
// .iter()
|
// i := 0
|
||||||
// .map(|&n| (
|
// loop if i == 10 break else {
|
||||||
// &self[n].kind,
|
// arr[i] = 0
|
||||||
// self[n].inputs.iter().position(|&n| n == mul_lhs)
|
// i += 1
|
||||||
// ))
|
// }
|
||||||
// .collect::<Vec<_>>()
|
//
|
||||||
// );
|
// |||||
|
||||||
|
// VVVVV
|
||||||
|
//
|
||||||
|
// cursor := &arr[0] + 0
|
||||||
|
// end := &arr[0] + 10
|
||||||
|
// loop if cursor == end else {
|
||||||
|
// *cursor = 0
|
||||||
|
// i += 1
|
||||||
|
// }
|
||||||
|
|
||||||
// let init_shift = self.new_node(
|
debug_assert!(self[iter_cond].inputs.contains(&index));
|
||||||
// self[rhs].ty,
|
let iter_bound_index =
|
||||||
// K::BinOp { op: T::Mul },
|
self[iter_cond].inputs.iter().rposition(|&n| n != index).unwrap();
|
||||||
// [ctrl, phi_lhs, mul_rhs],
|
debug_assert_ne!(iter_bound_index, 0);
|
||||||
// tys,
|
let end_shift = self.new_node(
|
||||||
// );
|
self[rhs].ty,
|
||||||
// let init =
|
K::BinOp { op: T::Mul },
|
||||||
// self.new_node(ty, K::BinOp { op: T::Add }, [ctrl, lhs, init_shift], tys);
|
[ctrl, self[iter_cond].inputs[iter_bound_index], step],
|
||||||
// let new_value = self.new_node_nop(ty, K::Phi, [phi_ctrl, init, 0]);
|
tys,
|
||||||
// let next = self.new_node(
|
);
|
||||||
// ty,
|
let end =
|
||||||
// Kind::BinOp { op: T::Add },
|
self.new_node(ty, K::BinOp { op: T::Add }, [ctrl, lhs, end_shift], tys);
|
||||||
// [ctrl, new_value, mul_rhs],
|
|
||||||
// tys,
|
let init_shift = self.new_node(
|
||||||
// );
|
self[rhs].ty,
|
||||||
// return Some(self.modify_input(new_value, 2, next));
|
K::BinOp { op: T::Mul },
|
||||||
//}
|
[ctrl, index_init, step],
|
||||||
|
tys,
|
||||||
|
);
|
||||||
|
let init =
|
||||||
|
self.new_node(ty, K::BinOp { op: T::Add }, [ctrl, lhs, init_shift], tys);
|
||||||
|
|
||||||
|
let new_value = self.new_node_nop(ty, K::Phi, [iter_loop, init, 0]);
|
||||||
|
let next =
|
||||||
|
self.new_node(ty, Kind::BinOp { op: T::Add }, [ctrl, new_value, step], tys);
|
||||||
|
|
||||||
|
let mut new_cond_inputs = self[iter_cond].inputs.clone();
|
||||||
|
new_cond_inputs[iter_bound_index] = end;
|
||||||
|
new_cond_inputs[3 - iter_bound_index] = new_value;
|
||||||
|
let new_cond = self.new_node(ty, self[iter_cond].kind, new_cond_inputs, tys);
|
||||||
|
self.replace(iter_cond, new_cond);
|
||||||
|
|
||||||
|
return Some(self.modify_input(new_value, 2, next));
|
||||||
|
}
|
||||||
|
|
||||||
if changed {
|
if changed {
|
||||||
return Some(self.new_node(ty, self[target].kind, [ctrl, lhs, rhs], tys));
|
return Some(self.new_node(ty, self[target].kind, [ctrl, lhs, rhs], tys));
|
||||||
|
@ -1268,6 +1293,14 @@ impl Nodes {
|
||||||
return Some(lhs);
|
return Some(lhs);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO: travese the graph downward and chech if this phi is only consumed by it
|
||||||
|
// self
|
||||||
|
if self[target].outputs.as_slice() == [rhs]
|
||||||
|
&& self[rhs].outputs.as_slice() == [target]
|
||||||
|
{
|
||||||
|
return Some(lhs);
|
||||||
|
}
|
||||||
|
|
||||||
if self[lhs].kind == Kind::Stre
|
if self[lhs].kind == Kind::Stre
|
||||||
&& self[rhs].kind == Kind::Stre
|
&& self[rhs].kind == Kind::Stre
|
||||||
&& self[lhs].ty == self[rhs].ty
|
&& self[lhs].ty == self[rhs].ty
|
||||||
|
|
|
@ -1,36 +1,35 @@
|
||||||
main:
|
main:
|
||||||
ADDI64 r254, r254, -104d
|
ADDI64 r254, r254, -88d
|
||||||
ST r31, r254, 48a, 56h
|
ST r31, r254, 48a, 40h
|
||||||
LRA r32, r0, :glob_stru
|
LRA r32, r0, :glob_stru
|
||||||
JAL r31, r0, :new_stru
|
JAL r31, r0, :new_stru
|
||||||
ST r1, r32, 0a, 16h
|
ST r1, r32, 0a, 16h
|
||||||
CP r33, r0
|
LD r33, r32, 0a, 8h
|
||||||
LD r34, r32, 0a, 8h
|
JEQ r33, r0, :0
|
||||||
JEQ r34, r33, :0
|
|
||||||
LI64 r32, 300d
|
LI64 r32, 300d
|
||||||
CP r1, r32
|
CP r1, r32
|
||||||
JMP :1
|
JMP :1
|
||||||
0: ST r33, r32, 0a, 8h
|
0: ST r0, r32, 0a, 8h
|
||||||
LD r34, r32, 0a, 8h
|
LD r33, r32, 0a, 8h
|
||||||
JEQ r34, r33, :2
|
JEQ r33, r0, :2
|
||||||
LI64 r32, 200d
|
LI64 r32, 200d
|
||||||
CP r1, r32
|
CP r1, r32
|
||||||
JMP :1
|
JMP :1
|
||||||
2: LI64 r34, 1d
|
2: LI64 r34, 1d
|
||||||
ST r34, r32, 0a, 8h
|
ST r34, r32, 0a, 8h
|
||||||
ST r34, r32, 8a, 8h
|
ST r34, r32, 8a, 8h
|
||||||
ADDI64 r35, r254, 0d
|
ADDI64 r33, r254, 0d
|
||||||
ST r34, r254, 0a, 8h
|
ST r34, r254, 0a, 8h
|
||||||
ST r34, r254, 8a, 8h
|
ST r34, r254, 8a, 8h
|
||||||
ST r34, r254, 16a, 8h
|
ST r34, r254, 16a, 8h
|
||||||
ST r34, r254, 24a, 8h
|
ST r34, r254, 24a, 8h
|
||||||
ST r34, r254, 32a, 8h
|
ST r34, r254, 32a, 8h
|
||||||
ST r34, r254, 40a, 8h
|
ST r34, r254, 40a, 8h
|
||||||
LI64 r36, 3d
|
ADDI64 r35, r33, 48d
|
||||||
CP r32, r33
|
CP r32, r33
|
||||||
8: JNE r32, r36, :3
|
8: JNE r35, r32, :3
|
||||||
LD r32, r254, 32a, 8h
|
LD r32, r254, 32a, 8h
|
||||||
JEQ r32, r33, :4
|
JEQ r32, r0, :4
|
||||||
LI64 r32, 100d
|
LI64 r32, 100d
|
||||||
CP r1, r32
|
CP r1, r32
|
||||||
JMP :1
|
JMP :1
|
||||||
|
@ -41,28 +40,25 @@ main:
|
||||||
ST r34, r254, 32a, 8h
|
ST r34, r254, 32a, 8h
|
||||||
ST r34, r254, 40a, 8h
|
ST r34, r254, 40a, 8h
|
||||||
CP r32, r33
|
CP r32, r33
|
||||||
7: LD r37, r254, 32a, 8h
|
7: LD r34, r254, 32a, 8h
|
||||||
JNE r32, r36, :5
|
JNE r35, r32, :5
|
||||||
JEQ r37, r33, :6
|
JEQ r34, r0, :6
|
||||||
LI64 r32, 10d
|
LI64 r32, 10d
|
||||||
CP r1, r32
|
CP r1, r32
|
||||||
JMP :1
|
JMP :1
|
||||||
6: CP r1, r33
|
6: CP r1, r0
|
||||||
JMP :1
|
JMP :1
|
||||||
5: MULI64 r37, r32, 16d
|
5: ADDI64 r34, r32, 16d
|
||||||
ADD64 r37, r35, r37
|
ST r0, r32, 0a, 8h
|
||||||
ST r33, r37, 0a, 8h
|
ST r0, r32, 8a, 8h
|
||||||
ST r33, r37, 8a, 8h
|
CP r32, r34
|
||||||
ADD64 r32, r32, r34
|
|
||||||
JMP :7
|
JMP :7
|
||||||
3: MULI64 r37, r32, 16d
|
3: JAL r31, r0, :new_stru
|
||||||
ADD64 r37, r35, r37
|
ST r1, r32, 0a, 16h
|
||||||
JAL r31, r0, :new_stru
|
ADDI64 r32, r32, 16d
|
||||||
ST r1, r37, 0a, 16h
|
|
||||||
ADD64 r32, r32, r34
|
|
||||||
JMP :8
|
JMP :8
|
||||||
1: LD r31, r254, 48a, 56h
|
1: LD r31, r254, 48a, 40h
|
||||||
ADDI64 r254, r254, 104d
|
ADDI64 r254, r254, 88d
|
||||||
JALA r0, r31, 0a
|
JALA r0, r31, 0a
|
||||||
new_stru:
|
new_stru:
|
||||||
ADDI64 r254, r254, -16d
|
ADDI64 r254, r254, -16d
|
||||||
|
@ -71,6 +67,6 @@ new_stru:
|
||||||
LD r1, r254, 0a, 16h
|
LD r1, r254, 0a, 16h
|
||||||
ADDI64 r254, r254, 16d
|
ADDI64 r254, r254, 16d
|
||||||
JALA r0, r31, 0a
|
JALA r0, r31, 0a
|
||||||
code size: 673
|
code size: 658
|
||||||
ret: 0
|
ret: 0
|
||||||
status: Ok(())
|
status: Ok(())
|
||||||
|
|
|
@ -1,31 +1,28 @@
|
||||||
main:
|
main:
|
||||||
ADDI64 r254, r254, -10240d
|
ADDI64 r254, r254, -10240d
|
||||||
LI64 r14, 1d
|
LI8 r15, 64b
|
||||||
LI8 r16, 64b
|
LI64 r16, 1024d
|
||||||
LI64 r17, 1024d
|
|
||||||
CP r13, r0
|
CP r13, r0
|
||||||
ADDI64 r18, r254, 0d
|
ADDI64 r17, r254, 0d
|
||||||
4: JLTU r13, r17, :0
|
4: JLTU r13, r16, :0
|
||||||
LI64 r16, 10d
|
ADDI64 r13, r17, 1024d
|
||||||
CP r13, r14
|
ADDI64 r15, r17, 10240d
|
||||||
3: LD r15, r254, 2048a, 1h
|
3: LD r14, r254, 2048a, 1h
|
||||||
JLTU r13, r16, :1
|
JLTU r13, r15, :1
|
||||||
ANDI r13, r15, 255d
|
ANDI r13, r14, 255d
|
||||||
CP r1, r13
|
CP r1, r13
|
||||||
JMP :2
|
JMP :2
|
||||||
1: ADD64 r15, r13, r14
|
1: ADDI64 r14, r13, 1024d
|
||||||
MULI64 r13, r13, 1024d
|
BMC r17, r13, 1024h
|
||||||
ADD64 r13, r18, r13
|
CP r13, r14
|
||||||
BMC r18, r13, 1024h
|
|
||||||
CP r13, r15
|
|
||||||
JMP :3
|
JMP :3
|
||||||
0: ADD64 r15, r13, r14
|
0: ADDI64 r14, r13, 1d
|
||||||
ADD64 r13, r18, r13
|
ADD64 r13, r17, r13
|
||||||
ST r16, r13, 0a, 1h
|
ST r15, r13, 0a, 1h
|
||||||
CP r13, r15
|
CP r13, r14
|
||||||
JMP :4
|
JMP :4
|
||||||
2: ADDI64 r254, r254, 10240d
|
2: ADDI64 r254, r254, 10240d
|
||||||
JALA r0, r31, 0a
|
JALA r0, r31, 0a
|
||||||
code size: 194
|
code size: 192
|
||||||
ret: 64
|
ret: 64
|
||||||
status: Ok(())
|
status: Ok(())
|
||||||
|
|
Loading…
Reference in a new issue