feat(bench): support for esp

laze-project.yml

@@ -652,6 +652,7 @@ modules:
       - nrf
       - rp
       - stm32
+      - esp
 
   - name: wifi-esp
     context:

src/riot-rs-bench/Cargo.toml

@@ -15,3 +15,6 @@ defmt = { workspace = true, optional = true }
 
 [target.'cfg(context = "cortex-m")'.dependencies]
 cortex-m = { workspace = true, features = ["critical-section-single-core"] }
+
+[target.'cfg(context = "esp")'.dependencies]
+esp-hal = { workspace = true }

src/riot-rs-bench/src/esp.rs

@@ -0,0 +1,28 @@
+use esp_hal::{
+    peripherals,
+    timer::systimer::{SystemTimer, Unit as _},
+};
+
+use crate::Error;
+
+#[allow(missing_docs)]
+pub fn benchmark<F: Fn() -> ()>(iterations: usize, f: F) -> Result<usize, Error> {
+    let mut systimer_periph = unsafe { peripherals::SYSTIMER::steal() };
+    let timer = SystemTimer::new(&mut systimer_periph);
+
+    // Reset counter of unit0, which is read in `SystemTimer::now()`.
+    timer.unit0.set_count(0);
+
+    while SystemTimer::now() == 0 {}
+
+    let before = SystemTimer::now();
+
+    for _ in 0..iterations {
+        f();
+    }
+
+    SystemTimer::now()
+        .checked_sub(before)
+        .map(|total| total as usize / iterations)
+        .ok_or(Error::SystemTimerWrapped)
+}

src/riot-rs-bench/src/lib.rs

@@ -9,8 +9,10 @@ cfg_if::cfg_if! {
     if #[cfg(context = "cortex-m")] {
         mod cortexm;
         use cortexm as bench;
-    }
-    else if #[cfg(context = "riot-rs")] {
+    } else if #[cfg(context = "esp")] {
+        mod esp;
+        use esp as bench;
+    } else if #[cfg(context = "riot-rs")] {
         // When run with laze but the architecture is not supported
         compile_error!("benchmarking is not supported for this architecture");
     } else {

tests/benchmarks/bench_sched_yield/laze.yml

@@ -2,4 +2,5 @@ apps:
   - name: bench_sched_yield
     selects:
       - sw/benchmark
+      - executor-thread
       - ?release