math.abs on complex (squared)

Percentage Accurate: 100.0% → 100.0%

Time: 1.8s

Alternatives: 2

Speedup: 1.0×

• 43.0% of points produce a very large (infinite) output. You may want to add a precondition.

Specification

Math

\[\begin{array}{l} \\ re \cdot re + im \cdot im \end{array} \]

FPCore

(FPCore modulus_sqr (re im) :precision binary64 (+ (* re re) (* im im)))

C

double modulus_sqr(double re, double im) {
	return (re * re) + (im * im);
}

Fortran

real(8) function modulus_sqr(re, im)
    real(8), intent (in) :: re
    real(8), intent (in) :: im
    modulus_sqr = (re * re) + (im * im)
end function

Java

public static double modulus_sqr(double re, double im) {
	return (re * re) + (im * im);
}

Python

def modulus_sqr(re, im):
	return (re * re) + (im * im)

Julia

function modulus_sqr(re, im)
	return Float64(Float64(re * re) + Float64(im * im))
end

MATLAB

function tmp = modulus_sqr(re, im)
	tmp = (re * re) + (im * im);
end

Wolfram

modulus$95$sqr[re_, im_] := N[(N[(re * re), $MachinePrecision] + N[(im * im), $MachinePrecision]), $MachinePrecision]

Initial Program: 100.0% accurate, 1.0× speedup

Math

\[\begin{array}{l} \\ re \cdot re + im \cdot im \end{array} \]

FPCore

(FPCore modulus_sqr (re im) :precision binary64 (+ (* re re) (* im im)))

C

double modulus_sqr(double re, double im) {
	return (re * re) + (im * im);
}

Fortran

real(8) function modulus_sqr(re, im)
    real(8), intent (in) :: re
    real(8), intent (in) :: im
    modulus_sqr = (re * re) + (im * im)
end function

Java

public static double modulus_sqr(double re, double im) {
	return (re * re) + (im * im);
}

Python

def modulus_sqr(re, im):
	return (re * re) + (im * im)

Julia

function modulus_sqr(re, im)
	return Float64(Float64(re * re) + Float64(im * im))
end

MATLAB

function tmp = modulus_sqr(re, im)
	tmp = (re * re) + (im * im);
end

Wolfram

modulus$95$sqr[re_, im_] := N[(N[(re * re), $MachinePrecision] + N[(im * im), $MachinePrecision]), $MachinePrecision]

Alternative 1: 100.0% accurate, 0.0× speedup

Math

\[\begin{array}{l} im_m = \left|im\right| \\ [re, im_m] = \mathsf{sort}([re, im_m])\\ \\ \mathsf{fma}\left(im_m, im_m, {re}^{2}\right) \end{array} \]

FPCore

im_m = (fabs.f64 im)
NOTE: re and im_m should be sorted in increasing order before calling this function.
(FPCore modulus_sqr (re im_m) :precision binary64 (fma im_m im_m (pow re 2.0)))

C

im_m = fabs(im);
assert(re < im_m);
double modulus_sqr(double re, double im_m) {
	return fma(im_m, im_m, pow(re, 2.0));
}

Julia

im_m = abs(im)
re, im_m = sort([re, im_m])
function modulus_sqr(re, im_m)
	return fma(im_m, im_m, (re ^ 2.0))
end

Wolfram

im_m = N[Abs[im], $MachinePrecision]
NOTE: re and im_m should be sorted in increasing order before calling this function.
modulus$95$sqr[re_, im$95$m_] := N[(im$95$m * im$95$m + N[Power[re, 2.0], $MachinePrecision]), $MachinePrecision]

Derivation

1. Initial program 100.0%

   \[re \cdot re + im \cdot im \]

2. Taylor expanded in re around 0 100.0%

   \[\leadsto \color{blue}{{im}^{2} + {re}^{2}} \]
3. Step-by-step derivation

   1. unpow2 100.0%

      \[\leadsto \color{blue}{im \cdot im} + {re}^{2} \]

   2. fma-udef 100.0%

      \[\leadsto \color{blue}{\mathsf{fma}\left(im, im, {re}^{2}\right)} \]

4. Simplified 100.0%

   \[\leadsto \color{blue}{\mathsf{fma}\left(im, im, {re}^{2}\right)} \]

5. Final simplification 100.0%

   \[\leadsto \mathsf{fma}\left(im, im, {re}^{2}\right) \]

Alternative 2: 100.0% accurate, 1.0× speedup

Math

\[\begin{array}{l} im_m = \left|im\right| \\ [re, im_m] = \mathsf{sort}([re, im_m])\\ \\ re \cdot re + im_m \cdot im_m \end{array} \]

FPCore

im_m = (fabs.f64 im)
NOTE: re and im_m should be sorted in increasing order before calling this function.
(FPCore modulus_sqr (re im_m) :precision binary64 (+ (* re re) (* im_m im_m)))

C

im_m = fabs(im);
assert(re < im_m);
double modulus_sqr(double re, double im_m) {
	return (re * re) + (im_m * im_m);
}

Fortran

im_m = abs(im)
NOTE: re and im_m should be sorted in increasing order before calling this function.
real(8) function modulus_sqr(re, im_m)
    real(8), intent (in) :: re
    real(8), intent (in) :: im_m
    modulus_sqr = (re * re) + (im_m * im_m)
end function

Java

im_m = Math.abs(im);
assert re < im_m;
public static double modulus_sqr(double re, double im_m) {
	return (re * re) + (im_m * im_m);
}

Python

im_m = math.fabs(im)
[re, im_m] = sort([re, im_m])
def modulus_sqr(re, im_m):
	return (re * re) + (im_m * im_m)

Julia

im_m = abs(im)
re, im_m = sort([re, im_m])
function modulus_sqr(re, im_m)
	return Float64(Float64(re * re) + Float64(im_m * im_m))
end

MATLAB

im_m = abs(im);
re, im_m = num2cell(sort([re, im_m])){:}
function tmp = modulus_sqr(re, im_m)
	tmp = (re * re) + (im_m * im_m);
end

Wolfram

im_m = N[Abs[im], $MachinePrecision]
NOTE: re and im_m should be sorted in increasing order before calling this function.
modulus$95$sqr[re_, im$95$m_] := N[(N[(re * re), $MachinePrecision] + N[(im$95$m * im$95$m), $MachinePrecision]), $MachinePrecision]

Derivation

1. Initial program 100.0%

   \[re \cdot re + im \cdot im \]

2. Final simplification 100.0%

   \[\leadsto re \cdot re + im \cdot im \]

Reproduce

herbie shell --seed 2023334 
(FPCore modulus_sqr (re im)
  :name "math.abs on complex (squared)"
  :precision binary64
  (+ (* re re) (* im im)))