From bb7f09ea36d5ba02fe5648941de51aa5765d4010 Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Mon, 10 Feb 2025 10:52:15 +0200
Subject: feat(zkvms_host_io): Metrics output now contains the name of the zkvm

---
 zkvmLib.nix                           | 2 ++
 zkvms_host_io/input_macros/src/lib.rs | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/zkvmLib.nix b/zkvmLib.nix
index 7f0198b..b1b38d4 100644
--- a/zkvmLib.nix
+++ b/zkvmLib.nix
@@ -68,6 +68,7 @@ in {
 
     buildGuestPhase = ''
       export INPUTS_DIR="$PWD/guests/${guest}"
+      export ZKVM="${args.pname}"
 
       pushd zkvms/${args.pname}/guest
       runHook preBuildGuest
@@ -85,6 +86,7 @@ in {
 
     buildPhase = ''
       export INPUTS_DIR="$PWD/guests/${guest}"
+      export ZKVM="${args.pname}"
 
       pushd zkvms/${args.pname}/host
       runHook preBuild
diff --git a/zkvms_host_io/input_macros/src/lib.rs b/zkvms_host_io/input_macros/src/lib.rs
index 10ff43b..8d7c2fc 100644
--- a/zkvms_host_io/input_macros/src/lib.rs
+++ b/zkvms_host_io/input_macros/src/lib.rs
@@ -133,7 +133,7 @@ pub fn benchmarkable(item: TokenStream) -> TokenStream {
              }}
 
              if run_info.benchmarking {{
-                 let mut output = String::new();
+                 let mut output = format!("zkvm,{{}}\n", env!("ZKVM"));
 
                  let duration = *ends.last().unwrap() - *starts.first().unwrap();
                  let duration = if run_info.millis {{ duration.as_millis() }} else {{ duration.as_secs().into() }};
-- 
cgit v1.2.3