Update benchmarks

Signed-off-by: Simon Davies <[email protected]>

Author: simongdavies

src/hyperlight_host/benches/benchmarks.rs

@@ -208,8 +208,8 @@ fn sandbox_heap_size_benchmark(c: &mut Criterion) {
 fn guest_call_heap_size_benchmark(c: &mut Criterion) {
     let mut group = c.benchmark_group("guest_call_heap_sizes");
 
-    // Helper function to create sandbox with specific heap size and call a guest function
-    let create_sandbox_and_call = |heap_size_mb: Option<u64>| {
+    // Helper function to create sandbox with specific heap size
+    let create_sandbox_with_heap_size = |heap_size_mb: Option<u64>| {
         let path = simple_guest_as_string().unwrap();
         let config = if let Some(size_mb) = heap_size_mb {
             let mut config = SandboxConfiguration::default();
@@ -221,42 +221,67 @@ fn guest_call_heap_size_benchmark(c: &mut Criterion) {
 
         let uninit_sandbox =
             UninitializedSandbox::new(GuestBinary::FilePath(path), config).unwrap();
-        let mut sandbox = uninit_sandbox.evolve(Noop::default()).unwrap();
-        
-        // Call the Echo function to test performance with different heap sizes
-        sandbox
-            .call_guest_function_by_name::<String>("Echo", "hello\n".to_string())
-            .unwrap()
+        uninit_sandbox.evolve(Noop::default()).unwrap()
     };
 
     // Benchmark guest function call with default heap size
     group.bench_function("guest_call_default_heap", |b| {
-        b.iter(|| create_sandbox_and_call(None));
+        let mut sandbox = create_sandbox_with_heap_size(None);
+        b.iter(|| {
+            sandbox
+                .call_guest_function_by_name::<String>("Echo", "hello\n".to_string())
+                .unwrap()
+        });
     });
 
     // Benchmark guest function call with 50MB heap
     group.bench_function("guest_call_50mb_heap", |b| {
-        b.iter(|| create_sandbox_and_call(Some(50)));
+        let mut sandbox = create_sandbox_with_heap_size(Some(50));
+        b.iter(|| {
+            sandbox
+                .call_guest_function_by_name::<String>("Echo", "hello\n".to_string())
+                .unwrap()
+        });
     });
 
     // Benchmark guest function call with 100MB heap
     group.bench_function("guest_call_100mb_heap", |b| {
-        b.iter(|| create_sandbox_and_call(Some(100)));
+        let mut sandbox = create_sandbox_with_heap_size(Some(100));
+        b.iter(|| {
+            sandbox
+                .call_guest_function_by_name::<String>("Echo", "hello\n".to_string())
+                .unwrap()
+        });
     });
 
     // Benchmark guest function call with 250MB heap
     group.bench_function("guest_call_250mb_heap", |b| {
-        b.iter(|| create_sandbox_and_call(Some(250)));
+        let mut sandbox = create_sandbox_with_heap_size(Some(250));
+        b.iter(|| {
+            sandbox
+                .call_guest_function_by_name::<String>("Echo", "hello\n".to_string())
+                .unwrap()
+        });
     });
 
     // Benchmark guest function call with 500MB heap
     group.bench_function("guest_call_500mb_heap", |b| {
-        b.iter(|| create_sandbox_and_call(Some(500)));
+        let mut sandbox = create_sandbox_with_heap_size(Some(500));
+        b.iter(|| {
+            sandbox
+                .call_guest_function_by_name::<String>("Echo", "hello\n".to_string())
+                .unwrap()
+        });
     });
 
     // Benchmark guest function call with 995MB heap
     group.bench_function("guest_call_995mb_heap", |b| {
-        b.iter(|| create_sandbox_and_call(Some(995)));
+        let mut sandbox = create_sandbox_with_heap_size(Some(995));
+        b.iter(|| {
+            sandbox
+                .call_guest_function_by_name::<String>("Echo", "hello\n".to_string())
+                .unwrap()
+        });
     });
 
     group.finish();