status-im
/
nim-dagger
mirror of https://github.com/status-im/nim-dagger.git
2
0
Fork 0
Code Issues Projects Releases Wiki Activity

updating ark cli

This commit is contained in:
Jaremy Creechley 2024-05-02 22:39:11 +03:00
parent 2e9a638fe2
commit f1732d58c2
No known key found for this signature in database
GPG Key ID: 4E66FB67B21D3300
3 changed files with 78 additions and 49 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

99
benchmarks/codex_ark_prover_cli.nim
View File

@ -21,10 +21,11 @@ type CircuitFiles* = object
dir*: string
circName*: string
proc runArkCircom(args: CircuitArgs, files: CircuitFiles, proofInputs: ProofInputs[Poseidon2Hash]) =
proc runArkCircom(
args: CircuitArgs, files: CircuitFiles, proofInputs: ProofInputs[Poseidon2Hash]
) =
echo "Loading sample proof..."
var
circom = CircomCompat.init(
var circom = CircomCompat.init(
files.r1cs,
files.wasm,
files.zkey,
@ -71,7 +72,6 @@ proc printHelp() =
quit(1)
proc parseCliOptions(args: var CircuitArgs, files: var CircuitFiles) =
var argCtr: int = 0
template expectPath(val: string): string =
if val == "":
@ -90,31 +90,41 @@ proc parseCliOptions(args: var CircuitArgs, files: var CircuitFiles) =
# Switches
of cmdLongOption, cmdShortOption:
case key
of "h", "help" : printHelp()
of "d", "depth" : args.depth = parseInt(value)
of "N", "maxslots" : args.maxslots = parseInt(value)
of "h", "help":
printHelp()
of "d", "depth":
args.depth = parseInt(value)
of "N", "maxslots":
args.maxslots = parseInt(value)
# of "c", "cellsize" : args.cellsize = checkPowerOfTwo(parseInt(value),"cellSize")
# of "b", "blocksize" : args.blocksize = checkPowerOfTwo(parseInt(value),"blockSize")
of "n", "nsamples" : args.nsamples = parseInt(value)
of "e", "entropy" : args.entropy = parseInt(value)
of "n", "nsamples":
args.nsamples = parseInt(value)
of "e", "entropy":
args.entropy = parseInt(value)
# of "S", "seed" : args.seed = parseInt(value)
of "s", "nslots" : args.nslots = parseInt(value)
of "K", "ncells" : args.ncells = checkPowerOfTwo(parseInt(value),"nCells")
of "i", "index" : args.index = parseInt(value)
of "r1cs" : files.r1cs = value.expectPath()
of "wasm" : files.wasm = value.expectPath()
of "zkey" : files.zkey = value.expectPath()
of "inputs" : files.inputs = value.expectPath()
of "dir" : files.dir = value.expectPath()
of "name" : files.circName = value
of "s", "nslots":
args.nslots = parseInt(value)
of "K", "ncells":
args.ncells = checkPowerOfTwo(parseInt(value), "nCells")
of "i", "index":
args.index = parseInt(value)
of "r1cs":
files.r1cs = value.expectPath()
of "wasm":
files.wasm = value.expectPath()
of "zkey":
files.zkey = value.expectPath()
of "inputs":
files.inputs = value.expectPath()
of "dir":
files.dir = value.expectPath()
of "name":
files.circName = value
else:
echo "Unknown option: ", key
echo "use --help to get a list of options"
quit()
of cmdEnd:
discard
@ -131,13 +141,21 @@ proc run*() =
parseCliOptions(args, files)
let dir = if files.dir != "": files.dir else: getCurrentDir()
let dir =
if files.dir != "":
files.dir
else:
getCurrentDir()
if files.circName != "":
if files.r1cs == "": files.r1cs = dir / fmt"{files.circName}.r1cs"
if files.wasm == "": files.wasm = dir / fmt"{files.circName}.wasm"
if files.zkey == "": files.zkey = dir / fmt"{files.circName}.zkey"
if files.r1cs == "":
files.r1cs = dir / fmt"{files.circName}.r1cs"
if files.wasm == "":
files.wasm = dir / fmt"{files.circName}.wasm"
if files.zkey == "":
files.zkey = dir / fmt"{files.circName}.zkey"
if files.inputs == "": files.inputs = dir / fmt"input.json"
if files.inputs == "":
files.inputs = dir / fmt"input.json"
echo "Got file args: ", files
@ -161,20 +179,29 @@ proc run*() =
inputs: JsonNode = !JsonNode.parse(inputData)
# sets default values for these args
if args.depth == 0: args.depth = codextypes.DefaultMaxSlotDepth # maximum depth of the slot tree
if args.maxslots == 0: args.maxslots = 256 # maximum number of slots
if args.depth == 0:
args.depth = codextypes.DefaultMaxSlotDepth
# maximum depth of the slot tree
if args.maxslots == 0:
args.maxslots = 256
# maximum number of slots
# sets number of samples to take
if args.nsamples == 0: args.nsamples = 1 # number of samples to prove
if args.nsamples == 0:
args.nsamples = 1
# number of samples to prove
# overrides the input.json params
if args.entropy != 0: inputs["entropy"] = %($args.entropy)
if args.nslots != 0: inputs["nSlotsPerDataSet"] = % args.nslots
if args.index != 0: inputs["slotIndex"] = % args.index
if args.ncells != 0: inputs["nCellsPerSlot"] = % args.ncells
if args.entropy != 0:
inputs["entropy"] = %($args.entropy)
if args.nslots != 0:
inputs["nSlotsPerDataSet"] = %args.nslots
if args.index != 0:
inputs["slotIndex"] = %args.index
if args.ncells != 0:
inputs["nCellsPerSlot"] = %args.ncells
var
proofInputs = Poseidon2Hash.jsonToProofInput(inputs)
var proofInputs = Poseidon2Hash.jsonToProofInput(inputs)
echo "Got args: ", args
runArkCircom(args, files, proofInputs)

2
benchmarks/config.nims
View File

@ -6,6 +6,8 @@
on
--tlsEmulation:
off
--d:
release
# when not defined(chronicles_log_level):
# --define:"chronicles_log_level:NONE" # compile all log statements

12
benchmarks/utils.nim
View File

@ -34,7 +34,7 @@ template benchmark*(benchmarkName: string, blk: untyped) =
import std/math
func floorLog2* (x: int): int =
func floorLog2*(x: int): int =
var k = -1
var y = x
while (y > 0):
@ -42,13 +42,13 @@ func floorLog2* (x: int): int =
y = y shr 1
return k
func ceilingLog2* (x: int): int =
if (x==0):
func ceilingLog2*(x: int): int =
if (x == 0):
return -1
else:
return (floorLog2(x-1) + 1)
return (floorLog2(x - 1) + 1)
func checkPowerOfTwo*(x: int , what: string): int =
func checkPowerOfTwo*(x: int, what: string): int =
let k = ceilingLog2(x)
assert( x == 2^k, ("`" & what & "` is expected to be a power of 2") )
assert(x == 2 ^ k, ("`" & what & "` is expected to be a power of 2"))
return x