From 0d22b3f8da6d967985abe327444b9d7e78d93710 Mon Sep 17 00:00:00 2001
From: Nicholas Ward <npward@berkeley.edu>
Date: Fri, 15 Jul 2022 10:23:44 -0700
Subject: [PATCH] fix

---
 evm/src/cpu/kernel/asm/sha2.asm | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/evm/src/cpu/kernel/asm/sha2.asm b/evm/src/cpu/kernel/asm/sha2.asm
index 414a2604..f779e1ff 100644
--- a/evm/src/cpu/kernel/asm/sha2.asm
+++ b/evm/src/cpu/kernel/asm/sha2.asm
@@ -1,10 +1,13 @@
 global count_bits:
+    JUMPDEST
     // stack: n (assumed to be > 0)
     push 0
     // stack: 0, n
     swap1
     // stack: n, 0
+    %jump(count_bits_loop)
 count_bits_loop:
+    JUMPDEST
     // stack: k, bits
     %div2
     // stack: k//2, bits
@@ -20,6 +23,7 @@ count_bits_loop:
     // stack: bits
 
 global sha2_store:
+    JUMPDEST
     // stack: num_u256s, x[0], x[1], x[2], ... , x[num_u256s-1]
     dup1
     // stack: num_u256s, num_u256s, x[0], x[1], x[2], ... , x[num_u256s-1]