Fix doc, pin version, add test

Author: nicole-graus

.github/workflows/bench-vs-nightly.yml

@@ -33,10 +33,10 @@ jobs:
       - name: Install SP1 toolchain
         run: |
           export PATH="$HOME/.cargo/bin:$HOME/.sp1/bin:$PATH"
-          if ! cargo prove --version >/dev/null 2>&1; then
+          if ! cargo prove --version 2>/dev/null | grep -q "6.0.1"; then
             curl -L https://sp1up.succinct.xyz | bash
             export PATH="$HOME/.sp1/bin:$PATH"
-            sp1up
+            sp1up --version v6.0.1
           fi
           cargo prove --version
 

bin/cli/src/main.rs

@@ -136,7 +136,8 @@ enum Commands {
         #[arg(long)]
         cycles: bool,
 
-        /// Build traces and print total main-trace field elements (rows × columns summed across all tables)
+        /// Build traces and print total main-trace field elements (rows × columns summed across
+        /// all tables) and aux-trace field elements (committed EF columns × rows)
         #[arg(long)]
         elements: bool,
     },
@@ -160,7 +161,7 @@ enum Commands {
         time: bool,
     },
 
-    /// Count main-trace field elements (rows × columns, summed across all tables) without proving
+    /// Count main-trace and aux-trace field elements without proving
     CountElements {
         /// Path to the ELF file
         #[arg(value_parser, value_hint = ValueHint::FilePath)]

prover/src/tests/prove_elfs_tests.rs

@@ -1909,3 +1909,20 @@ fn test_addiw_neg_immediate() {
     let result = crate::prove_and_verify(&elf_bytes).expect("prove_and_verify failed");
     assert!(result, "addiw with negative immediate should verify");
 }
+
+/// Regression test: both main and aux field element counts must be nonzero for any real ELF.
+/// Guards against silent under-count if a table is added to `Traces` but not enumerated in
+/// `total_field_elements` or `total_auxiliary_field_elements`.
+#[test]
+fn test_count_elements_nonzero() {
+    let elf_bytes = crate::test_utils::asm_elf_bytes("addi_one");
+    let (main, aux) = crate::count_elements(&elf_bytes, &[]).expect("count_elements failed");
+    assert!(
+        main > 0,
+        "total_field_elements should be nonzero (got {main})"
+    );
+    assert!(
+        aux > 0,
+        "total_auxiliary_field_elements should be nonzero (got {aux})"
+    );
+}