Skip to content

Commit

Permalink
implement recursive layout
Browse files Browse the repository at this point in the history
  • Loading branch information
TAdev0 committed Sep 5, 2024
1 parent 4e05778 commit 5f88468
Show file tree
Hide file tree
Showing 2 changed files with 15 additions and 0 deletions.
4 changes: 4 additions & 0 deletions integration_tests/cairozero_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -288,6 +288,8 @@ func runPythonVm(testFilename, path string) (time.Duration, string, string, erro
// A file without this suffix will use the default ("plain") layout.
if strings.HasSuffix(testFilename, ".small.cairo") {
args = append(args, "--layout", "small")
} else if strings.HasSuffix(testFilename, ".recursive.cairo") {
args = append(args, "--layout", "recursive")
} else if strings.HasSuffix(testFilename, ".starknet_with_keccak.cairo") {
args = append(args, "--layout", "starknet_with_keccak")
}
Expand Down Expand Up @@ -321,6 +323,8 @@ func runVm(path string) (time.Duration, string, string, string, error) {
layout := "plain"
if strings.Contains(path, ".small") {
layout = "small"
} else if strings.Contains(path, ".recursive") {
layout = "recursive"
} else if strings.Contains(path, ".starknet_with_keccak") {
layout = "starknet_with_keccak"
}
Expand Down
11 changes: 11 additions & 0 deletions pkg/vm/builtins/layouts.go
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,15 @@ func getPlainLayout() Layout {
return Layout{Name: "plain", RcUnits: 16, Builtins: []LayoutBuiltin{}}
}

func getRecursiveLayout() Layout {
return Layout{Name: "small", RcUnits: 16, Builtins: []LayoutBuiltin{
{Runner: &Output{}, Builtin: starknet.Output},
{Runner: &Pedersen{ratio: 128}, Builtin: starknet.Pedersen},
{Runner: &RangeCheck{ratio: 8, RangeCheckNParts: 8}, Builtin: starknet.RangeCheck},
{Runner: &Bitwise{ratio: 8}, Builtin: starknet.Bitwise},
}}
}

func getStarknetWithKeccakLayout() Layout {
return Layout{Name: "starknet_with_keccak", RcUnits: 4, Builtins: []LayoutBuiltin{
{Runner: &Output{}, Builtin: starknet.Output},
Expand All @@ -56,6 +65,8 @@ func GetLayout(layout string) (Layout, error) {
return getSmallLayout(), nil
case "plain":
return getPlainLayout(), nil
case "recursive":
return getRecursiveLayout(), nil
case "starknet_with_keccak":
return getStarknetWithKeccakLayout(), nil
case "":
Expand Down

0 comments on commit 5f88468

Please sign in to comment.