After global procedures and types are generated, then queue the generation of the procedures for each module

1 files changed, 35 insertions, 38 deletions

diff --git a/src/llvm_backend.cpp b/src/llvm_backend.cpp
index 1524f6c25..7360bdaa7 100644
--- a/src/llvm_backend.cpp
+++ b/src/llvm_backend.cpp
@@ -2061,9 +2061,26 @@ gb_internal lbProcedure *lb_create_cleanup_runtime(lbModule *main_module) { // C
 	return p;
 }
 
+gb_internal void lb_generate_procedure(lbModule *m, lbProcedure *p);
 
-gb_internal WORKER_TASK_PROC(lb_generate_procedures_and_types_per_module) {
+gb_internal WORKER_TASK_PROC(lb_generate_procedures_worker_proc) {
 	lbModule *m = cast(lbModule *)data;
+	for (isize i = 0; i < m->procedures_to_generate.count; i++) {
+		lbProcedure *p = m->procedures_to_generate[i];
+		lb_generate_procedure(p->module, p);
+	}
+	return 0;
+}
+
+
+struct lbGenerateProceduresWorkerData {
+	lbModule *m;
+	bool do_threading;
+};
+
+gb_internal WORKER_TASK_PROC(lb_generate_procedures_and_types_per_module) {
+	lbGenerateProceduresWorkerData *wd = cast(lbGenerateProceduresWorkerData *)data;
+	lbModule *m = wd->m;
 	for (Entity *e : m->global_types_to_create) {
 		(void)lb_get_entity_name(m, e);
 		(void)lb_type(m, e->type);
@@ -2073,6 +2090,10 @@ gb_internal WORKER_TASK_PROC(lb_generate_procedures_and_types_per_module) {
 		(void)lb_get_entity_name(m, e);
 		array_add(&m->procedures_to_generate, lb_create_procedure(m, e));
 	}
+
+	if (wd->do_threading) {
+		thread_pool_add_task(lb_generate_procedures_worker_proc, m);
+	}
 	return 0;
 }
 
@@ -2162,13 +2183,22 @@ gb_internal void lb_create_global_procedures_and_types(lbGenerator *gen, Checker
 
 	if (do_threading) {
 		for (auto const &entry : gen->modules) {
-			lbModule *m = entry.value;
-			thread_pool_add_task(lb_generate_procedures_and_types_per_module, m);
+			auto wd = permanent_alloc_item<lbGenerateProceduresWorkerData>();
+			wd->m = entry.value;
+			wd->do_threading = true;
+			thread_pool_add_task(lb_generate_procedures_and_types_per_module, wd);
 		}
 	} else {
 		for (auto const &entry : gen->modules) {
+			auto wd = permanent_alloc_item<lbGenerateProceduresWorkerData>();
+			wd->m = entry.value;
+			wd->do_threading = false;
+			lb_generate_procedures_and_types_per_module(wd);
+		}
+
+		for (auto const &entry : gen->modules) {
 			lbModule *m = entry.value;
-			lb_generate_procedures_and_types_per_module(m);
+			lb_generate_procedures_worker_proc(m);
 		}
 
 	}
@@ -2176,9 +2206,6 @@ gb_internal void lb_create_global_procedures_and_types(lbGenerator *gen, Checker
 	thread_pool_wait();
 }
 
-gb_internal void lb_generate_procedure(lbModule *m, lbProcedure *p);
-
-
 gb_internal bool lb_is_module_empty(lbModule *m) {
 	if (LLVMGetFirstFunction(m->mod) == nullptr &&
 	    LLVMGetFirstGlobal(m->mod) == nullptr) {
@@ -2395,33 +2422,6 @@ gb_internal WORKER_TASK_PROC(lb_llvm_module_pass_worker_proc) {
 	return 0;
 }
 
-
-
-gb_internal WORKER_TASK_PROC(lb_generate_procedures_worker_proc) {
-	lbModule *m = cast(lbModule *)data;
-	for (isize i = 0; i < m->procedures_to_generate.count; i++) {
-		lbProcedure *p = m->procedures_to_generate[i];
-		lb_generate_procedure(p->module, p);
-	}
-	return 0;
-}
-
-gb_internal void lb_generate_procedures(lbGenerator *gen, bool do_threading) {
-	if (do_threading) {
-		for (auto const &entry : gen->modules) {
-			lbModule *m = entry.value;
-			thread_pool_add_task(lb_generate_procedures_worker_proc, m);
-		}
-
-		thread_pool_wait();
-	} else {
-		for (auto const &entry : gen->modules) {
-			lbModule *m = entry.value;
-			lb_generate_procedures_worker_proc(m);
-		}
-	}
-}
-
 gb_internal WORKER_TASK_PROC(lb_generate_missing_procedures_to_check_worker_proc) {
 	lbModule *m = cast(lbModule *)data;
 	for (isize i = 0; i < m->missing_procedures_to_check.count; i++) {
@@ -3332,12 +3332,9 @@ gb_internal bool lb_generate_code(lbGenerator *gen) {
 		do_threading = false;
 	}
 
-	TIME_SECTION("LLVM Global Procedures and Types");
+	TIME_SECTION("LLVM Global Procedures and Types + Procedure Generation");
 	lb_create_global_procedures_and_types(gen, info, do_threading);
 
-	TIME_SECTION("LLVM Procedure Generation");
-	lb_generate_procedures(gen, do_threading);
-
 	if (build_context.command_kind == Command_test && !already_has_entry_point) {
 		TIME_SECTION("LLVM main");
 		lb_create_main_procedure(default_module, gen->startup_runtime, gen->cleanup_runtime);