style: use the '# =>' stack-comment convention in the batch kernel masm

Match the operand-stack (# =>) and advice-stack (# AS =>) comment style used
throughout the rest of the protocol assembly.

Author: claude

crates/miden-protocol/asm/kernels/batch/lib/note_tracker.masm

@@ -22,39 +22,39 @@ proc absorb_scratch_into_batch_hasher
     push.0 exec.memory::set_scratch_word_index
 
     exec.memory::load_batch_hasher_state
-    # Stack: [RATE0, RATE1, CAPACITY]
+    # => [RATE0, RATE1, CAPACITY]
 
     exec.memory::get_scratch_word_index
     exec.memory::get_scratch_words_count
     u32lt
-    # Stack: [should_loop, RATE0, RATE1, CAPACITY]
+    # => [should_loop, RATE0, RATE1, CAPACITY]
 
     while.true
         exec.memory::get_scratch_word_index
         mul.4 add.TX_NOTES_SCRATCH_PTR
-        # Stack: [scratch_ptr, RATE0, RATE1, CAPACITY]
+        # => [scratch_ptr, RATE0, RATE1, CAPACITY]
 
         padw dup.4 mem_loadw_le
-        # Stack: [DATA1, scratch_ptr, RATE0, RATE1, CAPACITY]
+        # => [DATA1, scratch_ptr, RATE0, RATE1, CAPACITY]
 
         padw dup.8 add.4 mem_loadw_le
-        # Stack: [DATA2, DATA1, scratch_ptr, RATE0, RATE1, CAPACITY]
+        # => [DATA2, DATA1, scratch_ptr, RATE0, RATE1, CAPACITY]
 
         movup.8 drop
-        # Stack: [DATA2, DATA1, RATE0, RATE1, CAPACITY]
+        # => [DATA2, DATA1, RATE0, RATE1, CAPACITY]
 
         # Replace RATE0 + RATE1 with DATA1 + DATA2 (DATA1 first, DATA2 second), then permute.
         swapdw
         dropw dropw
         swapw
         exec.poseidon2::permute
-        # Stack: [RATE0', RATE1', CAPACITY']
+        # => [RATE0', RATE1', CAPACITY']
 
         exec.memory::get_scratch_word_index add.2 exec.memory::set_scratch_word_index
         exec.memory::get_scratch_word_index
         exec.memory::get_scratch_words_count
         u32lt
-        # Stack: [should_loop, RATE0, RATE1, CAPACITY]
+        # => [should_loop, RATE0, RATE1, CAPACITY]
     end
 
     exec.memory::save_batch_hasher_state
@@ -87,51 +87,51 @@ pub proc compute_input_notes_commitment
 
     exec.memory::get_num_transactions
     push.0
-    # Stack: [tx_index, num_transactions]
+    # => [tx_index, num_transactions]
 
     dup.1 dup.1 neq
-    # Stack: [should_loop, tx_index, num_transactions]
+    # => [should_loop, tx_index, num_transactions]
 
     while.true
         dup exec.memory::get_tx_input_notes_commitment
-        # Stack: [INPUT_NOTES_COMMITMENT_i, tx_index, num_transactions]
+        # => [INPUT_NOTES_COMMITMENT_i, tx_index, num_transactions]
 
         dupw exec.word::eqz
-        # Stack: [is_empty, INPUT_NOTES_COMMITMENT_i, tx_index, num_transactions]
+        # => [is_empty, INPUT_NOTES_COMMITMENT_i, tx_index, num_transactions]
 
         if.true
             dropw
         else
             push.TX_NOTES_SCRATCH_PTR movdn.4
-            # Stack: [INPUT_NOTES_COMMITMENT_i, tx_notes_scratch_ptr, tx_index, num_transactions]
+            # => [INPUT_NOTES_COMMITMENT_i, tx_notes_scratch_ptr, tx_index, num_transactions]
 
             adv.push_mapvaln
             adv_push div.4
-            # Stack: [num_words, INPUT_NOTES_COMMITMENT_i, tx_notes_scratch_ptr, tx_index, num_transactions]
+            # => [num_words, INPUT_NOTES_COMMITMENT_i, tx_notes_scratch_ptr, tx_index, num_transactions]
 
             dup exec.memory::set_scratch_words_count
 
             movup.5 swap
-            # Stack: [num_words, tx_notes_scratch_ptr, INPUT_NOTES_COMMITMENT_i, tx_index, num_transactions]
+            # => [num_words, tx_notes_scratch_ptr, INPUT_NOTES_COMMITMENT_i, tx_index, num_transactions]
 
             # Pipe the note data into scratch, asserting its poseidon2 hash equals
             # INPUT_NOTES_COMMITMENT_i.
             exec.mem::pipe_preimage_to_memory
-            # Stack: [end_ptr, tx_index, num_transactions]
+            # => [end_ptr, tx_index, num_transactions]
             drop
 
             exec.absorb_scratch_into_batch_hasher
         end
-        # Stack: [tx_index, num_transactions]
+        # => [tx_index, num_transactions]
 
         add.1
         dup.1 dup.1 neq
-        # Stack: [should_loop, tx_index, num_transactions]
+        # => [should_loop, tx_index, num_transactions]
     end
 
     drop drop
 
     exec.memory::load_batch_hasher_state
     exec.poseidon2::squeeze_digest
-    # Stack: [INPUT_NOTES_COMMITMENT]
+    # => [INPUT_NOTES_COMMITMENT]
 end

crates/miden-protocol/asm/kernels/batch/lib/prologue.masm

@@ -44,59 +44,59 @@ pub proc prepare_batch
     # Layer 1: pipe BATCH_ID's mapped value to tx_tuples_ptr + verify.
     # ---------------------------------------------------------------------------------------------
 
-    # Stack: [BATCH_ID]
+    # => [BATCH_ID]
     push.TX_TUPLES_PTR movdn.4
-    # Stack: [BATCH_ID, tx_tuples_ptr]
+    # => [BATCH_ID, tx_tuples_ptr]
 
     adv.push_mapvaln
-    # AS: [len_felts, data...]
+    # AS => [len_felts, data...]
 
     adv_push div.4
-    # Stack: [num_words, BATCH_ID, tx_tuples_ptr]
+    # => [num_words, BATCH_ID, tx_tuples_ptr]
 
     # num_transactions = num_words / 2 (each tx contributes 2 words: tx_id + account_id_pair).
     dup div.2 exec.memory::set_num_transactions
 
     movup.5 swap
-    # Stack: [num_words, tx_tuples_ptr, BATCH_ID]
+    # => [num_words, tx_tuples_ptr, BATCH_ID]
 
     # Pipe the tuples into memory, asserting their poseidon2 hash equals BATCH_ID.
     exec.mem::pipe_preimage_to_memory
-    # Stack: [end_ptr]
+    # => [end_ptr]
     drop
 
     # Layer 2: for each transaction, pipe + verify its header.
     # ---------------------------------------------------------------------------------------------
 
     exec.memory::get_num_transactions
     push.0
-    # Stack: [tx_index, num_transactions]
+    # => [tx_index, num_transactions]
 
     dup.1 dup.1 neq
-    # Stack: [should_loop, tx_index, num_transactions]
+    # => [should_loop, tx_index, num_transactions]
 
     while.true
         dup exec.memory::get_tx_id
-        # Stack: [TX_ID, tx_index, num_transactions]
+        # => [TX_ID, tx_index, num_transactions]
 
         dup.4 exec.memory::tx_header_ptr movdn.4
-        # Stack: [TX_ID, tx_header_ptr, tx_index, num_transactions]
+        # => [TX_ID, tx_header_ptr, tx_index, num_transactions]
 
         adv.push_mapvaln
         adv_push div.4
-        # Stack: [num_words, TX_ID, tx_header_ptr, tx_index, num_transactions]
+        # => [num_words, TX_ID, tx_header_ptr, tx_index, num_transactions]
 
         movup.5 swap
-        # Stack: [num_words, tx_header_ptr, TX_ID, tx_index, num_transactions]
+        # => [num_words, tx_header_ptr, TX_ID, tx_index, num_transactions]
 
         # Pipe the header into memory, asserting its poseidon2 hash equals TX_ID.
         exec.mem::pipe_preimage_to_memory
-        # Stack: [end_ptr, tx_index, num_transactions]
+        # => [end_ptr, tx_index, num_transactions]
         drop
 
         add.1
         dup.1 dup.1 neq
-        # Stack: [should_loop, tx_index, num_transactions]
+        # => [should_loop, tx_index, num_transactions]
     end
 
     drop drop

crates/miden-protocol/asm/kernels/batch/main.masm

@@ -52,24 +52,24 @@ use miden::batch_kernel::prologue
 #! TODO: aggregate per-account updates and emit a separate ACCOUNT_UPDATES_COMMITMENT output.
 #! TODO: recursively verify each transaction's `ExecutionProof`.
 proc main
-    # Stack: [BLOCK_COMMITMENT, BATCH_ID, pad(8)]
+    # => [BLOCK_COMMITMENT, BATCH_ID, pad(8)]
 
     # TODO: verify BLOCK_COMMITMENT against block header data via the pipe-and-verify pattern.
     dropw
-    # Stack: [BATCH_ID, pad(12)]
+    # => [BATCH_ID, pad(12)]
 
     exec.prologue::prepare_batch
-    # Stack: [pad(16)]
+    # => [pad(16)]
 
     exec.note_tracker::compute_input_notes_commitment
-    # Stack: [INPUT_NOTES_COMMITMENT, pad(16)]
+    # => [INPUT_NOTES_COMMITMENT, pad(16)]
 
     # Drop the surplus padding word so the stack holds exactly the 16-felt output region
     # [INPUT_NOTES_COMMITMENT, BATCH_NOTE_TREE_ROOT, batch_expiration_block_num, pad(7)]. The
     # BATCH_NOTE_TREE_ROOT and batch_expiration_block_num cells stay zero (wired up in follow-up
     # PRs).
     movupw.3 dropw
-    # Stack: [INPUT_NOTES_COMMITMENT, pad(12)]
+    # => [INPUT_NOTES_COMMITMENT, pad(12)]
 end
 
 begin