diff --git a/juno_samples/rodinia/srad/benches/srad_bench.rs b/juno_samples/rodinia/srad/benches/srad_bench.rs
index 23dc06434c4fa77a0ca8e9e608ef6e9718489185..bf0e4ad4e954aa2d0035eefec26d9233ca805550 100644
--- a/juno_samples/rodinia/srad/benches/srad_bench.rs
+++ b/juno_samples/rodinia/srad/benches/srad_bench.rs
@@ -1,7 +1,7 @@
 #![feature(concat_idents)]
 use criterion::{criterion_group, criterion_main, Criterion};
 
-use hercules_rt::{runner, HerculesImmBox, HerculesImmBoxTo, HerculesMutBox, HerculesMutBoxTo};
+use hercules_rt::{runner, HerculesMutBox, HerculesMutBoxTo};
 
 juno_build::juno!("srad");
 
@@ -11,35 +11,37 @@ fn srad_bench(c: &mut Criterion) {
     let mut group = c.benchmark_group("srad bench");
     group.sample_size(10);
 
-    let mut r = runner!(srad);
-    let niter = 30;
-    let lambda = 0.5;
-    let nrows = 2048;
-    let ncols = 2048;
-    let image = "data/image.pgm".to_string();
-    let Image {
-        image: image_ori,
-        max,
-        rows: image_ori_rows,
-        cols: image_ori_cols,
-    } = read_graphics(image);
-    let image = resize(&image_ori, image_ori_rows, image_ori_cols, nrows, ncols);
-    let mut image_h = HerculesMutBox::from(image.clone());
-    group.bench_function("srad bench large", |b| {
-        b.iter(|| {
-            async_std::task::block_on(async {
-                r.run(
-                    nrows as u64,
-                    ncols as u64,
-                    niter as u64,
-                    image_h.to(),
-                    max,
-                    lambda,
-                )
-                .await
-            });
-        })
-    });
+    let mut bench = |name, niter, nrows, ncols| {
+        let mut r = runner!(srad);
+        let lambda = 0.5;
+        let image = "data/image.pgm".to_string();
+        let Image {
+            image: image_ori,
+            max,
+            rows: image_ori_rows,
+            cols: image_ori_cols,
+        } = read_graphics(image);
+        let image = resize(&image_ori, image_ori_rows, image_ori_cols, nrows, ncols);
+        let mut image_h = HerculesMutBox::from(image.clone());
+        group.bench_function(name, |b| {
+            b.iter(|| {
+                async_std::task::block_on(async {
+                    r.run(
+                        nrows as u64,
+                        ncols as u64,
+                        niter as u64,
+                        image_h.to(),
+                        max,
+                        lambda,
+                    )
+                    .await
+                });
+            })
+        });
+    };
+
+    bench("srad bench small", 100, 512, 512);
+    bench("srad bench large", 30, 2048, 2048);
 }
 
 criterion_group!(benches, srad_bench);