diff --git a/src/llvm_backend.hpp b/src/llvm_backend.hpp index e69d3a6ed..8bd038c16 100644 --- a/src/llvm_backend.hpp +++ b/src/llvm_backend.hpp @@ -298,6 +298,7 @@ struct lbProcedure { lbBlock * entry_block; lbBlock * curr_block; lbTargetList * target_list; + PtrMap direct_parameters; Ast *curr_stmt; diff --git a/src/llvm_backend_general.cpp b/src/llvm_backend_general.cpp index 0dabee076..18cdd196d 100644 --- a/src/llvm_backend_general.cpp +++ b/src/llvm_backend_general.cpp @@ -2590,6 +2590,15 @@ lbValue lb_find_or_add_entity_string_byte_slice_with_type(lbModule *m, String co lbValue lb_find_ident(lbProcedure *p, lbModule *m, Entity *e, Ast *expr) { + if (e->flags & EntityFlag_Param) { + // NOTE(bill): Bypass the stack copied variable for + // direct parameters as there is no need for the direct load + auto *found = map_get(&p->direct_parameters, e); + if (found) { + return *found; + } + } + auto *found = map_get(&m->values, e); if (found) { auto v = *found; diff --git a/src/llvm_backend_proc.cpp b/src/llvm_backend_proc.cpp index a88c8f2ee..2347c6bc0 100644 --- a/src/llvm_backend_proc.cpp +++ b/src/llvm_backend_proc.cpp @@ -486,6 +486,8 @@ void lb_begin_procedure_body(lbProcedure *p) { p->entry_block = lb_create_block(p, "entry", true); lb_start_block(p, p->entry_block); + map_init(&p->direct_parameters, heap_allocator()); + GB_ASSERT(p->type != nullptr); lb_ensure_abi_function_type(p->module, p); @@ -539,6 +541,8 @@ void lb_begin_procedure_body(lbProcedure *p) { param.value = value; param.type = e->type; + map_set(&p->direct_parameters, e, param); + lbValue ptr = lb_address_from_load_or_generate_local(p, param); GB_ASSERT(LLVMIsAAllocaInst(ptr.value)); lb_add_entity(p->module, e, ptr);