feat: implement dex layout (#649)

* dex layout * fix
NethermindEth · Sep 5, 2024 · 4134714 · 4134714
1 parent 4e05778
commit 4134714
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 4 deletions.
diff --git a/integration_tests/cairozero_test.go b/integration_tests/cairozero_test.go
@@ -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, ".dex.cairo") {
+		args = append(args, "--layout", "dex")
 	} else if strings.HasSuffix(testFilename, ".starknet_with_keccak.cairo") {
 		args = append(args, "--layout", "starknet_with_keccak")
 	}
@@ -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, ".dex") {
+		layout = "dex"
 	} else if strings.Contains(path, ".starknet_with_keccak") {
 		layout = "starknet_with_keccak"
 	}

diff --git a/pkg/vm/builtins/layouts.go b/pkg/vm/builtins/layouts.go
@@ -24,6 +24,10 @@ type Layout struct {
 	Builtins []LayoutBuiltin
 }
 
+func getPlainLayout() Layout {
+	return Layout{Name: "plain", RcUnits: 16, Builtins: []LayoutBuiltin{}}
+}
+
 func getSmallLayout() Layout {
 	return Layout{Name: "small", RcUnits: 16, Builtins: []LayoutBuiltin{
 		{Runner: &Output{}, Builtin: starknet.Output},
@@ -33,8 +37,13 @@ func getSmallLayout() Layout {
 	}}
 }
 
-func getPlainLayout() Layout {
-	return Layout{Name: "plain", RcUnits: 16, Builtins: []LayoutBuiltin{}}
+func getDexLayout() Layout {
+	return Layout{Name: "dex", RcUnits: 4, Builtins: []LayoutBuiltin{
+		{Runner: &Output{}, Builtin: starknet.Output},
+		{Runner: &Pedersen{ratio: 8}, Builtin: starknet.Pedersen},
+		{Runner: &RangeCheck{ratio: 8, RangeCheckNParts: 8}, Builtin: starknet.RangeCheck},
+		{Runner: &ECDSA{ratio: 512}, Builtin: starknet.ECDSA},
+	}}
 }
 
 func getStarknetWithKeccakLayout() Layout {
@@ -52,10 +61,12 @@ func getStarknetWithKeccakLayout() Layout {
 
 func GetLayout(layout string) (Layout, error) {
 	switch layout {
-	case "small":
-		return getSmallLayout(), nil
 	case "plain":
 		return getPlainLayout(), nil
+	case "small":
+		return getSmallLayout(), nil
+	case "dex":
+		return getDexLayout(), nil
 	case "starknet_with_keccak":
 		return getStarknetWithKeccakLayout(), nil
 	case "":