Actually pass the allocator in so it's used instead of temp.

Codegen.jai

@@ -627,7 +627,7 @@ emit_declaration :: (state : *Codegen_State, node : *AST_Node) {
 	}
 }
 
-codegen :: (result : *Compiler_Context) {
+codegen :: (result : *Compiler_Context, allocator := temp) {
 	codegen(result, .HLSL);
 }
 

module.jai

@@ -507,10 +507,10 @@ compile_file :: (ctx : *Compiler_Context, path : string, allocator : Allocator =
 
 		ctx.file = make_file(ctx, path);
 
-		lex(ctx);
+		lex(ctx, allocator);
-		parse(ctx);
+		parse(ctx, allocator);
-		check(ctx);
+		check(ctx, allocator);
-		codegen(ctx);
+		codegen(ctx, allocator);
 		generate_output_data(ctx);
 	}
 }