From 510e0ec526a7b701d758d0c2ae46c8afbe1ae567 Mon Sep 17 00:00:00 2001
From: mtfishman <mfishman@flatironinstitute.org>
Date: Thu, 20 Jun 2024 14:02:51 -0400
Subject: [PATCH] Improve type stability of sqrt(::Complex)

---
 base/complex.jl | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/base/complex.jl b/base/complex.jl
index 8ac126d2c6532..8504f02860602 100644
--- a/base/complex.jl
+++ b/base/complex.jl
@@ -514,13 +514,13 @@ function robust_cinv(c::Float64, d::Float64)
 end
 
 function ssqs(x::T, y::T) where T<:Real
-    k::Int = 0
+    k = 0
     ρ = x*x + y*y
     if !isfinite(ρ) && (isinf(x) || isinf(y))
         ρ = convert(T, Inf)
     elseif isinf(ρ) || (ρ==0 && (x!=0 || y!=0)) || ρ<nextfloat(zero(T))/(2*eps(T)^2)
         m::T = max(abs(x), abs(y))
-        k = m==0 ? m : exponent(m)
+        k = m==0 ? k : exponent(m)
         xk, yk = ldexp(x,-k), ldexp(y,-k)
         ρ = xk*xk + yk*yk
     end
@@ -533,7 +533,7 @@ function sqrt(z::Complex)
     if x==y==0
         return Complex(zero(x),y)
     end
-    ρ, k::Int = ssqs(x, y)
+    ρ, k = ssqs(x, y)
     if isfinite(x) ρ=ldexp(abs(x),-k)+sqrt(ρ) end
     if isodd(k)
         k = div(k-1,2)