Add_all_cairo_stwo_layout

lambdaclass · Feb 16, 2025 · bc25090 · bc25090
1 parent 153d3a8
commit bc25090
Show file tree

Hide file tree

Showing 4 changed files with 38 additions and 0 deletions.
diff --git a/vm/src/types/instance_definitions/builtins_instance_def.rs b/vm/src/types/instance_definitions/builtins_instance_def.rs
@@ -179,6 +179,28 @@ impl BuiltinsInstanceDef {
         }
     }
 
+    pub(crate) fn all_cairo_stwo() -> BuiltinsInstanceDef {
+        BuiltinsInstanceDef {
+            output: true,
+            pedersen: Some(PedersenInstanceDef::new(Some(256))),
+            range_check: Some(RangeCheckInstanceDef::default()),
+            ecdsa: None,
+            bitwise: Some(BitwiseInstanceDef::new(Some(16))),
+            ec_op: None,
+            keccak: None,
+            poseidon: Some(PoseidonInstanceDef::new(Some(256))),
+            range_check96: Some(RangeCheckInstanceDef::new(Some(8))),
+            #[cfg(feature = "mod_builtin")]
+            add_mod: Some(ModInstanceDef::new(Some(128), 1, 96)),
+            #[cfg(feature = "mod_builtin")]
+            mul_mod: Some(ModInstanceDef::new(Some(256), 1, 96)),
+            #[cfg(not(feature = "mod_builtin"))]
+            add_mod: None,
+            #[cfg(not(feature = "mod_builtin"))]
+            mul_mod: None,
+        }
+    }
+
     pub(crate) fn all_solidity() -> BuiltinsInstanceDef {
         BuiltinsInstanceDef {
             output: true,

diff --git a/vm/src/types/layout.rs b/vm/src/types/layout.rs
@@ -132,6 +132,18 @@ impl CairoLayout {
         }
     }
 
+    pub(crate) fn all_cairo_stwo_instance() -> CairoLayout {
+        CairoLayout {
+            name: LayoutName::all_cairo_stwo,
+            rc_units: 4,
+            cpu_component_step: DEFAULT_CPU_COMPONENT_STEP,
+            memory_units_per_step: DEFAULT_MEMORY_UNITS_PER_STEP,
+            builtins: BuiltinsInstanceDef::all_cairo_stwo(),
+            public_memory_fraction: 8,
+            diluted_pool_instance_def: Some(DilutedPoolInstanceDef::default()),
+        }
+    }
+
     pub(crate) fn all_solidity_instance() -> CairoLayout {
         CairoLayout {
             name: LayoutName::all_solidity,

diff --git a/vm/src/types/layout_name.rs b/vm/src/types/layout_name.rs
@@ -21,6 +21,7 @@ pub enum LayoutName {
     all_solidity,
     all_cairo,
     dynamic,
+    all_cairo_stwo,
 }
 
 impl LayoutName {
@@ -37,6 +38,7 @@ impl LayoutName {
             LayoutName::all_solidity => "all_solidity",
             LayoutName::all_cairo => "all_cairo",
             LayoutName::dynamic => "dynamic",
+            LayoutName::all_cairo_stwo => "all_cairo_stwo",
         }
     }
 }
@@ -62,6 +64,7 @@ impl ValueEnum for LayoutName {
             Self::all_solidity,
             Self::all_cairo,
             Self::dynamic,
+            Self::all_cairo_stwo,
         ]
     }
 

diff --git a/vm/src/vm/runners/cairo_runner.rs b/vm/src/vm/runners/cairo_runner.rs
@@ -188,6 +188,7 @@ impl CairoRunner {
             LayoutName::recursive_large_output => CairoLayout::recursive_large_output_instance(),
             LayoutName::recursive_with_poseidon => CairoLayout::recursive_with_poseidon(),
             LayoutName::all_cairo => CairoLayout::all_cairo_instance(),
+            LayoutName::all_cairo_stwo => CairoLayout::all_cairo_stwo_instance(),
             LayoutName::all_solidity => CairoLayout::all_solidity_instance(),
             LayoutName::dynamic => {
                 let params =