ci: add benchmark config

Author: AvivYossef-starkware

crates/bench_tools/src/main.rs

@@ -1,3 +1,4 @@
+use bench_tools::types::benchmark_config::{find_benchmarks_by_package, BENCHMARKS};
 use clap::{Parser, Subcommand};
 
 #[derive(Parser)]
@@ -18,14 +19,45 @@ enum Commands {
         #[arg(short, long)]
         out: String,
     },
+    /// List benchmarks for a package.
+    List {
+        /// Package name to list benchmarks for. If not provided, lists all benchmarks.
+        #[arg(short, long)]
+        package: Option<String>,
+    },
 }
 
 fn main() {
     let cli = Cli::parse();
-
     match cli.command {
         Commands::Run { package: _, out: _ } => {
             unimplemented!()
         }
+        Commands::List { package } => match package {
+            Some(package_name) => {
+                let benchmarks = find_benchmarks_by_package(&package_name);
+
+                if benchmarks.is_empty() {
+                    println!("No benchmarks found for package: {}", package_name);
+                    return;
+                }
+
+                println!("Available benchmarks for package '{}':", package_name);
+                for bench in &benchmarks {
+                    println!("  - {} (runs: {})", bench.name, bench.cmd_args.join(" "));
+                }
+            }
+            None => {
+                println!("All available benchmarks:");
+                for bench in BENCHMARKS {
+                    println!(
+                        "  - {} (package: {}, runs: {})",
+                        bench.name,
+                        bench.package,
+                        bench.cmd_args.join(" ")
+                    );
+                }
+            }
+        },
     }
 }

crates/bench_tools/src/types.rs

@@ -1,3 +1,4 @@
+pub mod benchmark_config;
 pub mod estimates;
 #[cfg(test)]
 mod estimates_test;

crates/bench_tools/src/types/benchmark_config.rs

@@ -0,0 +1,42 @@
+/// Configuration for a single benchmark.
+#[derive(Debug, Clone)]
+pub struct BenchmarkConfig {
+    pub name: &'static str,
+    pub package: &'static str,
+    pub cmd_args: &'static [&'static str],
+}
+
+impl BenchmarkConfig {
+    /// Get the full cargo bench command as owned strings.
+    pub fn cmd_args_owned(&self) -> Vec<String> {
+        self.cmd_args.iter().map(|s| s.to_string()).collect()
+    }
+}
+
+/// All available benchmarks defined as a const array.
+pub const BENCHMARKS: &[BenchmarkConfig] = &[
+    BenchmarkConfig {
+        name: "full_committer_flow",
+        package: "starknet_committer_and_os_cli",
+        cmd_args: &["bench", "-p", "starknet_committer_and_os_cli", "full_committer_flow"],
+    },
+    BenchmarkConfig {
+        name: "single_tree_flow",
+        package: "starknet_committer_and_os_cli",
+        cmd_args: &["bench", "-p", "starknet_committer_and_os_cli", "single_tree_flow"],
+    },
+    BenchmarkConfig {
+        name: "gateway_apply_block",
+        package: "apollo_gateway",
+        cmd_args: &["bench", "-p", "apollo_gateway", "apply_block"],
+    },
+];
+
+/// Helper functions for working with benchmarks.
+pub fn find_benchmark_by_name(name: &str) -> Option<&'static BenchmarkConfig> {
+    BENCHMARKS.iter().find(|b| b.name == name)
+}
+
+pub fn find_benchmarks_by_package(package: &str) -> Vec<&'static BenchmarkConfig> {
+    BENCHMARKS.iter().filter(|b| b.package == package).collect()
+}