aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--zkvmLib.nix2
-rw-r--r--zkvms_host_io/input_macros/src/lib.rs2
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() }};