ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-07-23 16:07:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve array programming code generation

Browse Source
This commit is contained in:
gingerBill
2020-10-20 15:38:41 +01:00
parent f4b4cd0433
commit c4dbc88a12
2 changed files with 28 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/ir.cpp
+13 -6
View File
@@ -4495,18 +4495,25 @@ irValue *ir_emit_arith(irProcedure *proc, TokenKind op, irValue *left, irValue *
if (inline_array_arith) {
// inline
for (i32 i = 0; i < count; i++) {
irValue *x = ir_emit_load(proc, ir_emit_array_epi(proc, lhs, i));
irValue *y = ir_emit_load(proc, ir_emit_array_epi(proc, rhs, i));
irValue *x_ptr = ir_emit_array_epi(proc, lhs, i);
irValue *y_ptr = ir_emit_array_epi(proc, rhs, i);
irValue *dst_ptr = ir_emit_array_epi(proc, res, i);
irValue *x = ir_emit_load(proc, x_ptr);
irValue *y = ir_emit_load(proc, y_ptr);
irValue *z = ir_emit_arith(proc, op, x, y, elem_type);
ir_emit_store(proc, ir_emit_array_epi(proc, res, i), z);
ir_emit_store(proc, dst_ptr, z);
}
} else {
auto loop_data = ir_loop_start(proc, count, t_i32);
irValue *x = ir_emit_load(proc, ir_emit_array_ep(proc, lhs, loop_data.idx));
irValue *y = ir_emit_load(proc, ir_emit_array_ep(proc, rhs, loop_data.idx));
irValue *x_ptr = ir_emit_array_ep(proc, lhs, loop_data.idx);
irValue *y_ptr = ir_emit_array_ep(proc, rhs, loop_data.idx);
irValue *dst_ptr = ir_emit_array_ep(proc, res, loop_data.idx);
irValue *x = ir_emit_load(proc, x_ptr);
irValue *y = ir_emit_load(proc, y_ptr);
irValue *z = ir_emit_arith(proc, op, x, y, elem_type);
ir_emit_store(proc, ir_emit_array_ep(proc, res, loop_data.idx), z);
ir_emit_store(proc, dst_ptr, z);
ir_loop_end(proc, loop_data);
}
src/llvm_backend.cpp
+15 -7
View File
@@ -5496,18 +5496,26 @@ lbValue lb_emit_arith(lbProcedure *p, TokenKind op, lbValue lhs, lbValue rhs, Ty
if (inline_array_arith) {
for (i64 i = 0; i < count; i++) {
lbValue a = lb_emit_load(p, lb_emit_array_epi(p, x, i));
lbValue b = lb_emit_load(p, lb_emit_array_epi(p, y, i));
lbValue a_ptr = lb_emit_array_epi(p, x, i);
lbValue b_ptr = lb_emit_array_epi(p, y, i);
lbValue dst_ptr = lb_emit_array_epi(p, res.addr, i);
lbValue a = lb_emit_load(p, a_ptr);
lbValue b = lb_emit_load(p, b_ptr);
lbValue c = lb_emit_arith(p, op, a, b, elem_type);
lb_emit_store(p, lb_emit_array_epi(p, res.addr, i), c);
lb_emit_store(p, dst_ptr, c);
}
} else {
auto loop_data = lb_loop_start(p, count);
auto loop_data = lb_loop_start(p, count, t_i32);
lbValue a = lb_emit_load(p, lb_emit_array_ep(p, x, loop_data.idx));
lbValue b = lb_emit_load(p, lb_emit_array_ep(p, y, loop_data.idx));
lbValue a_ptr = lb_emit_array_ep(p, x, loop_data.idx);
lbValue b_ptr = lb_emit_array_ep(p, y, loop_data.idx);
lbValue dst_ptr = lb_emit_array_ep(p, res.addr, loop_data.idx);
lbValue a = lb_emit_load(p, a_ptr);
lbValue b = lb_emit_load(p, b_ptr);
lbValue c = lb_emit_arith(p, op, a, b, elem_type);
lb_emit_store(p, lb_emit_array_ep(p, res.addr, loop_data.idx), c);
lb_emit_store(p, dst_ptr, c);
lb_loop_end(p, loop_data);
}