google
diff --git a/‎extensions/math_ext.cc
Copy file name to clipboardExpand all lines: extensions/math_ext.cc
+15Lines changed: 15 additions & 0 deletions b/‎extensions/math_ext.cc
Copy file name to clipboardExpand all lines: extensions/math_ext.cc
+15Lines changed: 15 additions & 0 deletions
diff --git a/‎extensions/math_ext_decls.cc
Copy file name to clipboardExpand all lines: extensions/math_ext_decls.cc
+7Lines changed: 7 additions & 0 deletions b/‎extensions/math_ext_decls.cc
Copy file name to clipboardExpand all lines: extensions/math_ext_decls.cc
+7Lines changed: 7 additions & 0 deletions
diff --git a/‎extensions/math_ext_test.cc
Copy file name to clipboardExpand all lines: extensions/math_ext_test.cc
+6Lines changed: 6 additions & 0 deletions b/‎extensions/math_ext_test.cc
Copy file name to clipboardExpand all lines: extensions/math_ext_test.cc
+6Lines changed: 6 additions & 0 deletions
@@ -205,6 +205,12 @@ double RoundDouble(double value) { return std::round(value); }
 
 double TruncDouble(double value) { return std::trunc(value); }
 
+double SqrtDouble(double value) { return std::sqrt(value); }
+
+double SqrtInt(int64_t value) { return std::sqrt(value); }
+
+double SqrtUint(uint64_t value) { return std::sqrt(value); }
+
 bool IsInfDouble(double value) { return std::isinf(value); }
 
 bool IsNaNDouble(double value) { return std::isnan(value); }
@@ -364,6 +370,15 @@ absl::Status RegisterMathExtensionFunctions(FunctionRegistry& registry,
   CEL_RETURN_IF_ERROR(
       (UnaryFunctionAdapter<double, double>::RegisterGlobalOverload(
           "math.round", RoundDouble, registry)));
+  CEL_RETURN_IF_ERROR(
+      (UnaryFunctionAdapter<double, double>::RegisterGlobalOverload(
+          "math.sqrt", SqrtDouble, registry)));
+  CEL_RETURN_IF_ERROR(
+      (UnaryFunctionAdapter<double, int64_t>::RegisterGlobalOverload(
+          "math.sqrt", SqrtInt, registry)));
+  CEL_RETURN_IF_ERROR(
+      (UnaryFunctionAdapter<double, uint64_t>::RegisterGlobalOverload(
+          "math.sqrt", SqrtUint, registry)));
   CEL_RETURN_IF_ERROR(
       (UnaryFunctionAdapter<double, double>::RegisterGlobalOverload(
           "math.trunc", TruncDouble, registry)));
 
@@ -121,19 +121,26 @@ absl::Status AddMinMaxDecls(TypeCheckerBuilder& builder) {
 absl::Status AddSignednessDecls(TypeCheckerBuilder& builder) {
   const Type kNumerics[] = {IntType(), DoubleType(), UintType()};
 
+  FunctionDecl sqrt_decl;
+  sqrt_decl.set_name("math.sqrt");
+
   FunctionDecl sign_decl;
   sign_decl.set_name("math.sign");
 
   FunctionDecl abs_decl;
   abs_decl.set_name("math.abs");
 
   for (const Type& type : kNumerics) {
+    CEL_RETURN_IF_ERROR(sqrt_decl.AddOverload(
+        MakeOverloadDecl(absl::StrCat("math_sqrt_", OverloadTypeName(type)),
+                         DoubleType(), type)));
     CEL_RETURN_IF_ERROR(sign_decl.AddOverload(MakeOverloadDecl(
         absl::StrCat("math_sign_", OverloadTypeName(type)), type, type)));
     CEL_RETURN_IF_ERROR(abs_decl.AddOverload(MakeOverloadDecl(
         absl::StrCat("math_abs_", OverloadTypeName(type)), type, type)));
   }
 
+  CEL_RETURN_IF_ERROR(builder.AddFunction(sqrt_decl));
   CEL_RETURN_IF_ERROR(builder.AddFunction(sign_decl));
   CEL_RETURN_IF_ERROR(builder.AddFunction(abs_decl));
 
 
@@ -550,6 +550,12 @@ INSTANTIATE_TEST_SUITE_P(
          {"math.ceil(42.01) == 43.0"},
          {"math.floor(42.01) == 42.0"},
          {"math.round(42.5) == 43.0"},
+         {"math.sqrt(49.0) == 7.0"},
+         {"math.sqrt(0) == 0.0"},
+         {"math.sqrt(1) == 1.0"},
+         {"math.sqrt(25u) == 5.0"},
+         {"math.sqrt(38.44) == 6.2"},
+         {"math.isNaN(math.sqrt(-15)) == true"},
          {"math.trunc(42.0) == 42.0"},
          {"math.isInf(42.0 / 0.0) == true"},
          {"math.isNaN(double('nan')) == true"},