ec_op changes 0.11 release

src/vm/runners/builtin_runner/ec_op.rs

@@ -1,4 +1,6 @@
 use std::borrow::Cow;
+use std::cell::RefCell;
+use std::collections::HashMap;
 
 use crate::math_utils::{ec_add, ec_double, safe_div_usize};
 use crate::types::instance_definitions::ec_op_instance_def::{
@@ -27,6 +29,7 @@ pub struct EcOpBuiltinRunner {
     pub(crate) stop_ptr: Option<usize>,
     pub(crate) included: bool,
     instances_per_component: u32,
+    cache: RefCell<HashMap<Relocatable, Felt>>,
 }
 
 impl EcOpBuiltinRunner {
@@ -40,6 +43,7 @@ impl EcOpBuiltinRunner {
             stop_ptr: None,
             included,
             instances_per_component: 1,
+            cache: RefCell::new(HashMap::new()),
         }
     }
     ///Returns True if the point (x, y) is on the elliptic curve defined as
@@ -154,6 +158,13 @@ impl EcOpBuiltinRunner {
             return Ok(None);
         }
         let instance = Relocatable::from((address.segment_index, address.offset - index));
+        let x_addr = (instance + (&Felt::new(INPUT_CELLS_PER_EC_OP)))
+            .map_err(|_| RunnerError::Memory(MemoryError::ExpectedInteger(instance)))?;
+
+        if let Some(number) = self.cache.borrow().get(&address).cloned() {
+            return Ok(Some(MaybeRelocatable::Int(number)));
+        }
+
         //All input cells should be filled, and be integer values
         //If an input cell is not filled, return None
         let mut input_cells = Vec::<&Felt>::with_capacity(self.n_input_cells as usize);
@@ -205,6 +216,14 @@ impl EcOpBuiltinRunner {
             &prime,
             self.ec_op_builtin.scalar_height,
         )?;
+        self.cache
+            .borrow_mut()
+            .insert(x_addr, result.0.clone().into());
+        self.cache.borrow_mut().insert(
+            (x_addr + 1usize)
+                .map_err(|_| RunnerError::Memory(MemoryError::ExpectedInteger(x_addr)))?,
+            result.1.clone().into(),
+        );
         match index - self.n_input_cells as usize {
             0 => Ok(Some(MaybeRelocatable::Int(Felt::new(result.0)))),
             _ => Ok(Some(MaybeRelocatable::Int(Felt::new(result.1)))),