From 79c02c8fd6e1236ba14fdd1eefac688b54878d3c Mon Sep 17 00:00:00 2001
From: samkim-crypto <skim13@cs.stanford.edu>
Date: Wed, 30 Aug 2023 11:16:09 -0700
Subject: [PATCH] [zk-token-proof] Add CUs to close account instruction
 (#33042)

add compute units for close context state account
---
 programs/zk-token-proof/src/lib.rs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/programs/zk-token-proof/src/lib.rs b/programs/zk-token-proof/src/lib.rs
index d350e3a897..e52f6520d3 100644
--- 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)
         }