From 601d190f3401c71864399a40df32bb8c9ace42dd Mon Sep 17 00:00:00 2001
From: Kamen Mladenov <kamen@syndamia.com>
Date: Tue, 11 Feb 2025 17:09:27 +0200
Subject: feat(zkvms_host_io): Output guest name when benchmarking

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

diff --git a/zkvmLib.nix b/zkvmLib.nix
index b1b38d4..8e4cb0b 100644
--- a/zkvmLib.nix
+++ b/zkvmLib.nix
@@ -68,7 +68,7 @@ in {
 
     buildGuestPhase = ''
       export INPUTS_DIR="$PWD/guests/${guest}"
-      export ZKVM="${args.pname}"
+      export ZKVM="${args.pname}" GUEST="${guest}"
 
       pushd zkvms/${args.pname}/guest
       runHook preBuildGuest
@@ -86,7 +86,7 @@ in {
 
     buildPhase = ''
       export INPUTS_DIR="$PWD/guests/${guest}"
-      export ZKVM="${args.pname}"
+      export ZKVM="${args.pname}" GUEST="${guest}"
 
       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 465a9b4..c3a34f7 100644
--- a/zkvms_host_io/input_macros/src/lib.rs
+++ b/zkvms_host_io/input_macros/src/lib.rs
@@ -197,7 +197,7 @@ pub fn benchmarkable(item: TokenStream) -> TokenStream {
              }}
 
              if run_info.benchmarking {{
-                 let mut output = format!("zkvm,{{}}\n", env!("ZKVM"));
+                 let mut output = format!("zkvm,{{}}\nguest,{{}}\n", env!("ZKVM"), env!("GUEST"));
 
                  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