From 6c3ea3721375f83629d381b50db4c1b4fdbfd842 Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Fri, 7 Feb 2025 11:56:48 +0200
Subject: feat(zkvms/jolt): Add benchmarking capability

---
 zkvms/jolt/host/src/main.rs | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'zkvms/jolt')

diff --git a/zkvms/jolt/host/src/main.rs b/zkvms/jolt/host/src/main.rs
index fee201f..b47966c 100644
--- a/zkvms/jolt/host/src/main.rs
+++ b/zkvms/jolt/host/src/main.rs
@@ -1,4 +1,4 @@
-use zkvms_host_io::{read_args, RunType::{ Execute, Prove, Verify }};
+use zkvms_host_io::{benchmarkable, read_args, RunType::{ Execute, Prove, Verify }};
 
 type Input = (Vec<Vec<bool>>, u32, Vec<Vec<u32>>);
 
@@ -13,12 +13,12 @@ pub fn main() {
 
     match run_info.run_type {
         Execute => unreachable!(),
-        Prove => {
-            let (output, _) = prove_guest(run_info.input.into());
+        Prove => benchmarkable!{
+            let (output, _) = prove_guest(run_info.input.clone().into());
             println!("Prove output: {}", output);
         },
-        Verify => {
-            let (_, proof) = prove_guest(run_info.input.into());
+        Verify => benchmarkable!{
+            let (_, proof) = prove_guest(run_info.input.clone().into());
             let is_valid = verify_guest(proof);
             println!("Verify is valid: {}", is_valid);
         },
-- 
cgit v1.2.3