reclaim stack space used in init

1 files changed, 10 insertions, 2 deletions

diff --git a/macros/src/codegen.rs b/macros/src/codegen.rs
index cf728a7d..21db1438 100644
--- a/macros/src/codegen.rs
+++ b/macros/src/codegen.rs
@@ -66,9 +66,17 @@ pub fn app(app: &App, analysis: &Analysis, extra: &Extra) -> TokenStream2 {
 
                 #(#pre_init_stmts)*
 
-                #call_init
+                #[inline(never)]
+                fn __rtic_init_resources<F>(f: F) where F: FnOnce() {
+                    f();
+                }
+
+                // Wrap late_init_stmts in a function to ensure that stack space is reclaimed.
+                __rtic_init_resources(||{
+                    #call_init
 
-                #(#post_init_stmts)*
+                    #(#post_init_stmts)*
+                });
 
                 #call_idle
             }