Use stack when the list is smaller than a given value (#1144)

Signed-off-by: James Sturtevant <[email protected]>
bytecodealliance · Jan 30, 2025 · cae738d · cae738d
1 parent b19ef10
commit cae738d
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/crates/csharp/src/function.rs b/crates/csharp/src/function.rs
@@ -853,13 +853,27 @@ impl Bindgen for FunctionBindgen<'_, '_> {
                 //TODO: wasm64
                 let align = self.interface_gen.csharp_gen.sizes.align(element).align_wasm32();
 
+                let (array_size, element_type) = crate::world_generator::dotnet_aligned_array(
+                    size,
+                    align,
+                );
+                let ret_area = self.locals.tmp("retArea");
+
                 self.needs_cleanup = true;
                 uwrite!(
                     self.src,
                     "
-                    var {buffer_size} = {size} * (nuint){list}.Count;
-                    var {address} = NativeMemory.AlignedAlloc({buffer_size}, {align});
-                    cleanups.Add(()=> NativeMemory.AlignedFree({address}));
+                    void* {address};
+                    if (({size} * {list}.Count) < 1024) {{
+                        var {ret_area} = stackalloc {element_type}[({array_size}*{list}.Count)+1];
+                        {address} = (void*)(((int){ret_area}) + ({align} - 1) & -{align});
+                    }} 
+                    else
+                    {{
+                        var {buffer_size} = {size} * (nuint){list}.Count;
+                        {address} = NativeMemory.AlignedAlloc({buffer_size}, {align});
+                        cleanups.Add(()=> NativeMemory.AlignedFree({address}));
+                    }}
 
                     for (int {index} = 0; {index} < {list}.Count; ++{index}) {{
                         {ty} {block_element} = {list}[{index}];