From 87be9776484546f8291c5bcdfef182666db8d6cd Mon Sep 17 00:00:00 2001
From: Zhang Zhuo <mycinbrin@gmail.com>
Date: Mon, 23 Sep 2024 16:03:41 +0800
Subject: [PATCH] improve err msg

---
 crates/core/machine/src/cpu/trace.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/crates/core/machine/src/cpu/trace.rs b/crates/core/machine/src/cpu/trace.rs
index 8f6eec515d..6fbf1a1726 100644
--- a/crates/core/machine/src/cpu/trace.rs
+++ b/crates/core/machine/src/cpu/trace.rs
@@ -548,7 +548,8 @@ impl CpuChip {
     fn pad_to_power_of_two<F: PrimeField32>(&self, shape: &Option<CoreShape>, values: &mut Vec<F>) {
         let n_real_rows = values.len() / NUM_CPU_COLS;
         let padded_nb_rows = if let Some(shape) = shape {
-            1 << shape.inner[&MachineAir::<F>::name(self)]
+            let name = MachineAir::<F>::name(self);
+            1 << shape.inner.get(&name).expect(&format!("fail to get shape of {}", name))
         } else if n_real_rows < 16 {
             16
         } else {