feat(zkvms_host_io): Metrics output now contains the name of the zkvm

2 files changed, 3 insertions, 1 deletions

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() }};