From 7a39154e536aec2d2cefc89104ebd85358ddfc15 Mon Sep 17 00:00:00 2001
From: FrancoGiachetta <francogiachetta27@gmail.com>
Date: Thu, 24 Oct 2024 12:49:23 -0300
Subject: [PATCH 1/3] add u128 square root

---
 src/vm/uint128.rs | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/src/vm/uint128.rs b/src/vm/uint128.rs
index 7dc4c83..86c71c5 100644
--- a/src/vm/uint128.rs
+++ b/src/vm/uint128.rs
@@ -1,5 +1,5 @@
 use super::EvalAction;
-use crate::Value;
+use crate::{debug::debug_signature, Value};
 use cairo_lang_sierra::{
     extensions::{
         core::{CoreLibfunc, CoreType},
@@ -24,7 +24,7 @@ pub fn eval(
     match selector {
         Uint128Concrete::Const(info) => eval_const(registry, info, args),
         Uint128Concrete::Operation(info) => eval_operation(registry, info, args),
-        Uint128Concrete::SquareRoot(_) => todo!(),
+        Uint128Concrete::SquareRoot(info) => eval_square_root(registry, info, args),
         Uint128Concrete::Equal(info) => eval_equal(registry, info, args),
         Uint128Concrete::ToFelt252(info) => eval_to_felt(registry, info, args),
         Uint128Concrete::FromFelt252(info) => eval_from_felt(registry, info, args),
@@ -54,6 +54,23 @@ pub fn eval_guarantee_mul(
     EvalAction::NormalBranch(0, smallvec![high, low, Value::Unit])
 }
 
+pub fn eval_square_root(
+    _registry: &ProgramRegistry<CoreType, CoreLibfunc>,
+    _info: &SignatureOnlyConcreteLibfunc,
+    args: Vec<Value>,
+) -> EvalAction {
+    let [range_check @ Value::Unit, Value::U128(value)]: [Value; 2] = args.try_into().unwrap()
+    else {
+        panic!()
+    };
+
+    let value_big = BigUint::from(value);
+
+    let result: u64 = value_big.sqrt().try_into().unwrap();
+
+    EvalAction::NormalBranch(0, smallvec![range_check, Value::U64(result)])
+}
+
 pub fn eval_guarantee_verify(
     _registry: &ProgramRegistry<CoreType, CoreLibfunc>,
     _info: &SignatureOnlyConcreteLibfunc,

From 208e20ba9829d638b4cb8be7516e859d1b73fc0e Mon Sep 17 00:00:00 2001
From: FrancoGiachetta <francogiachetta27@gmail.com>
Date: Fri, 25 Oct 2024 15:57:43 -0300
Subject: [PATCH 2/3] Squash comparison

---
 src/value.rs      |  4 +++-
 src/vm/uint128.rs | 24 ++++++++++++++++++++++++
 2 files changed, 27 insertions(+), 1 deletion(-)

diff --git a/src/value.rs b/src/value.rs
index c8729e4..88acacb 100644
--- a/src/value.rs
+++ b/src/value.rs
@@ -182,7 +182,9 @@ impl Value {
             CoreTypeConcrete::Nullable(info) => self.is(registry, &info.ty),
             CoreTypeConcrete::Uninitialized(_) => matches!(self, Self::Uninitialized { .. }),
             CoreTypeConcrete::Felt252DictEntry(_) => todo!(),
-            CoreTypeConcrete::SquashedFelt252Dict(_) => todo!(),
+            CoreTypeConcrete::SquashedFelt252Dict(info) => {
+                matches!(self, Self::FeltDict { ty, .. } if *ty == info.ty)
+            },
             CoreTypeConcrete::Pedersen(_) => matches!(self, Self::Unit),
             CoreTypeConcrete::Poseidon(_) => matches!(self, Self::Unit),
             CoreTypeConcrete::Span(_) => todo!(),
diff --git a/src/vm/uint128.rs b/src/vm/uint128.rs
index 86c71c5..fe35766 100644
--- a/src/vm/uint128.rs
+++ b/src/vm/uint128.rs
@@ -232,3 +232,27 @@ pub fn eval_byte_reverse(
 
     EvalAction::NormalBranch(0, smallvec![bitwise, Value::U128(value)])
 }
+
+#[cfg(test)]
+mod test {
+    use crate::{load_cairo, test_utils::run_test_program, Value};
+
+    #[test]
+    fn test_square_root() {
+        let (_, program) = load_cairo!(
+            use core::num::traits::Sqrt;
+            fn main() -> u64 {
+                0xffffffffffffffffffffffffffffffff_u128.sqrt()
+            }
+        );
+
+        let result = run_test_program(program);
+
+        let Value::U64(payload) = result.last().unwrap()
+        else {
+            panic!("No output");
+        };
+
+        assert_eq!(*payload, 0xffffffffffffffff);
+    }
+}

From ae9eae66b5d983238b231a99bc8a4604384123a6 Mon Sep 17 00:00:00 2001
From: FrancoGiachetta <francogiachetta27@gmail.com>
Date: Tue, 29 Oct 2024 11:18:07 -0300
Subject: [PATCH 3/3] match Felt252DictEntry

---
 src/value.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/value.rs b/src/value.rs
index 88acacb..1b477cd 100644
--- a/src/value.rs
+++ b/src/value.rs
@@ -181,7 +181,7 @@ impl Value {
             CoreTypeConcrete::Sint64(_) => todo!(),
             CoreTypeConcrete::Nullable(info) => self.is(registry, &info.ty),
             CoreTypeConcrete::Uninitialized(_) => matches!(self, Self::Uninitialized { .. }),
-            CoreTypeConcrete::Felt252DictEntry(_) => todo!(),
+            CoreTypeConcrete::Felt252DictEntry(info) => matches!(self, Self::FeltDictEntry { ty, .. } if *ty == info.ty),
             CoreTypeConcrete::SquashedFelt252Dict(info) => {
                 matches!(self, Self::FeltDict { ty, .. } if *ty == info.ty)
             },