add forgotten allocation stats clear

fluencelabs · Dec 6, 2023 · 46e957a · 46e957a
1 parent f158def
commit 46e957a
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/marine/src/marine.rs b/marine/src/marine.rs
@@ -157,9 +157,13 @@ impl<WB: WasmBackend> Marine<WB> {
             *cp = call_parameters;
         }
 
-        self.core
+        let result = self.core
             .call(module_name, func_name, args)
-            .map_err(|e| check_for_oom_and_convert_error(&self.core, e))
+            .map_err(|e| check_for_oom_and_convert_error(&self.core, e))?;
+
+        self.core.clear_allocation_stats();
+
+        Ok(result)
     }
 
     /// Call a specified function of loaded on a startup module by its name.