blob: 942b49f9935f6cae736b52ea436e5678eab362a5 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
{
description = "zkVMs benchmarks";
nixConfig = {
extra-substituters = "https://nix-blockchain-development.cachix.org";
extra-trusted-public-keys = "nix-blockchain-development.cachix.org-1:Ekei3RuW3Se+P/UIo6Q/oAgor/fVhFuuuX5jR8K/cdg=";
};
inputs = {
mcl-blockchain.url = "github:metacraft-labs/nix-blockchain-development?ref=zkvm-packages";
nixpkgs.follows = "mcl-blockchain/nixpkgs";
crane.follows = "mcl-blockchain/crane";
rust-overlay.follows = "mcl-blockchain/rust-overlay";
# flake-utils.follows = "mcl-blockchain/flake-utils";
};
outputs = { self, nixpkgs, mcl-blockchain, crane, rust-overlay, ... }:
let
system = "x86_64-linux";
pkgs = import nixpkgs { system = system; overlays = [
mcl-blockchain.overlays.default
rust-overlay.overlays.default
];
};
callPackage = pkgs.lib.callPackageWith pkgs;
fixDeps = commonArgs: commonArgs // {
postUnpack = ''
${commonArgs.postUnpack or ""}
ln -s ../../../guests ./source/zkvms/${commonArgs.pname}/guest/
ln -s ../../../guests_macro ./source/zkvms/${commonArgs.pname}/guest/
ln -s ../../Cargo.lock ./source/zkvms/${commonArgs.pname}/
'';
preBuild = ''
${commonArgs.preBuild or ""}
cd zkvms/${commonArgs.pname}
'';
};
# Creates custom build and install phases
# Adds the "buildGuest" phase
# Adds the "run" pseudo-phase (running your zkVM is done with a shell script,
# this "phase" allows one to add things to the script)
# Requirements:
# - zkVM is inside zkvms/pname/
# - guest crate is located at zkvms/pname/guest and is named "guest"
withCustomPhases = guest: currentPackage: let
hostBin = currentPackage.hostBin or ("host-" + currentPackage.pname);
in with currentPackage; {
phases = [
"unpackPhase" "patchPhase" "configurePhase" # Standard phases
"buildGuestPhase" # Custom phase
"buildPhase" "checkPhase" "installPhase" "fixupPhase" # Standard phases
];
buildGuestPhase = ''
pushd zkvms/${currentPackage.pname}/guest
runHook preBuildGuest
${currentPackage.buildGuestCommand or "cargo build --release"} \
${if currentPackage ? guestTarget then "--target " + currentPackage.guestTarget else ""} \
${currentPackage.guestExtraArgs or ""}
${if currentPackage ? guestTarget then "ln -s ../../guest/target/${currentPackage.guestTarget}/release/guest ../host/src/guest" else ""}
unset RUSTUP_TOOLCHAIN RUSTFLAGS CARGO_ENCODED_RUSTFLAGS
runHook postBuildGuest
popd
'';
buildPhase = ''
export INPUTS="$PWD/Vertices-010.in"
pushd zkvms/${currentPackage.pname}
runHook preBuild
cargo build --bin ${hostBin} --release
runHook postBuild
popd
'';
installPhase = let
preRunBinaries =
if currentPackage ? preRunBinaries && builtins.length currentPackage.preRunBinaries > 0 then
"export PATH=\"\\$PATH:" + pkgs.lib.makeBinPath currentPackage.preRunBinaries + "\""
else
"";
preRunLibraries =
if currentPackage ? preRunLibraries && builtins.length currentPackage.preRunLibraries > 0 then
"export LD_LIBRARY_PATH=\"\\$LD_LIBRARY_PATH:" + pkgs.lib.makeLibraryPath currentPackage.preRunLibraries + "\""
else
"";
in ''
runHook preInstall
mkdir -p "$out"/bin
for bin in $(find . -type f -regex "./zkvms/.*release/[^/]*" -executable -print)
do
mv "$bin" "$out"/bin/
done
cat <<EOF > "$out"/bin/${pname}
#!/usr/bin/env sh
${preRunBinaries}
${preRunLibraries}
${currentPackage.preRun or ""}
"$out"/bin/${hostBin} \$@
EOF
chmod +x "$out"/bin/${pname}
runHook postInstall
'';
doNotPostBuildInstallCargoBinaries = true;
} // currentPackage;
createPackages = guestName: let
guest = if guestName == null then "graph_coloring" else guestName;
postfix = if guestName == null then "" else "/" + guest;
args-zkVM = {
craneLib-default = crane.mkLib pkgs;
zkVM-helpers = {
inherit fixDeps;
withCustomPhases = withCustomPhases guest;
};
};
in {
"risc0${postfix}" = callPackage ./zkvms/risc0/default.nix args-zkVM;
"sp1${postfix}" = callPackage ./zkvms/sp1/default.nix args-zkVM;
"zkwasm${postfix}" = callPackage ./zkvms/zkwasm/default.nix args-zkVM;
"zkm${postfix}" = callPackage ./zkvms/zkm/default.nix args-zkVM;
"jolt${postfix}" = callPackage ./zkvms/jolt/default.nix args-zkVM;
"nexus${postfix}" = callPackage ./zkvms/nexus/default.nix args-zkVM;
};
guests = [ null ] ++ (builtins.attrNames
(pkgs.lib.filterAttrs
(_: type: type == "directory")
(builtins.readDir ./guests)));
in {
packages.${system} = pkgs.lib.foldr
(guest: accum: accum // (createPackages guest))
{}
guests;
};
}
|
