Chore: make pipeline benchmarks less susceptible to noise (#5502)

Signed-off-by: Connor Tsui <[email protected]>

Author: connortsui20

vortex/benches/pipeline.rs

@@ -137,15 +137,13 @@ const T: usize = 32;
 const S: usize = N * W / T;
 
 /// Benchmark sizes to test for performance benchmarks.
-const BENCHMARK_SIZES: [usize; 8] = [
+const BENCHMARK_SIZES: [usize; 6] = [
     1024,   // 1K
     8192,   // 8K
     16384,  // 16K
     65536,  // 64K
-    73728,  // 72K
     86016,  // 84K
     100352, // 98K
-    262144, // 256K
 ];
 
 /// Sizes to test for correctness verification.
@@ -793,7 +791,7 @@ fn compare_outputs(function_name: &str, expected: &[f32], actual: &[f32], expect
 // Benchmarks
 ////////////////////////////////////////////////////////////////////////////////////////////////////
 
-#[divan::bench(consts = BENCHMARK_SIZES)]
+#[divan::bench(consts = BENCHMARK_SIZES, sample_size = 512)]
 fn batch<const SIZE: usize>(bencher: Bencher) {
     let (input_data, mut buffers) = setup(SIZE);
 
@@ -812,7 +810,7 @@ fn batch<const SIZE: usize>(bencher: Bencher) {
     });
 }
 
-#[divan::bench(consts = BENCHMARK_SIZES)]
+#[divan::bench(consts = BENCHMARK_SIZES, sample_size = 512)]
 fn pipeline<const SIZE: usize>(bencher: Bencher) {
     let (input_data, mut buffers) = setup(SIZE);
     bencher.bench_local(|| {
@@ -830,7 +828,7 @@ fn pipeline<const SIZE: usize>(bencher: Bencher) {
     });
 }
 
-#[divan::bench(consts = BENCHMARK_SIZES)]
+#[divan::bench(consts = BENCHMARK_SIZES, sample_size = 512)]
 fn pipeline_extra_copy<const SIZE: usize>(bencher: Bencher) {
     let (input_data, mut buffers) = setup(SIZE);
     bencher.bench_local(|| {
@@ -849,7 +847,7 @@ fn pipeline_extra_copy<const SIZE: usize>(bencher: Bencher) {
     });
 }
 
-#[divan::bench(consts = BENCHMARK_SIZES)]
+#[divan::bench(consts = BENCHMARK_SIZES, sample_size = 512)]
 fn in_place_batch<const SIZE: usize>(bencher: Bencher) {
     let (input_data, mut buffers) = setup(SIZE);
     bencher.bench_local(|| {
@@ -865,7 +863,7 @@ fn in_place_batch<const SIZE: usize>(bencher: Bencher) {
     });
 }
 
-#[divan::bench(consts = BENCHMARK_SIZES)]
+#[divan::bench(consts = BENCHMARK_SIZES, sample_size = 512)]
 fn in_place_pipeline<const SIZE: usize>(bencher: Bencher) {
     let (input_data, mut buffers) = setup(SIZE);
     bencher.bench_local(|| {