[zk-token-proof] Add CUs to close account instruction (#33042)

add compute units for close context state account
solana-labs · Aug 30, 2023 · 79c02c8 · 79c02c8
1 parent 3458a7d
commit 79c02c8
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/programs/zk-token-proof/src/lib.rs b/programs/zk-token-proof/src/lib.rs
@@ -134,6 +134,11 @@ declare_process_instruction!(process_instruction, 0, |invoke_context| {
 
     match instruction {
         ProofInstruction::CloseContextState => {
+            if native_programs_consume_cu {
+                invoke_context
+                    .consume_checked(3_300)
+                    .map_err(|_| InstructionError::ComputationalBudgetExceeded)?;
+            }
             ic_msg!(invoke_context, "CloseContextState");
             process_close_proof_context(invoke_context)
         }