math.square on complex, real part

Percentage Accurate: 93.2% → 98.0%

Time: 2.7s

Alternatives: 3

Speedup: 0.6×

• 44.2% 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 re_sqr (re im) :precision binary64 (- (* re re) (* im im)))

C

double re_sqr(double re, double im) {
	return (re * re) - (im * im);
}

Fortran

real(8) function re_sqr(re, im)
    real(8), intent (in) :: re
    real(8), intent (in) :: im
    re_sqr = (re * re) - (im * im)
end function

Java

public static double re_sqr(double re, double im) {
	return (re * re) - (im * im);
}

Python

def re_sqr(re, im):
	return (re * re) - (im * im)

Julia

function re_sqr(re, im)
	return Float64(Float64(re * re) - Float64(im * im))
end

MATLAB

function tmp = re_sqr(re, im)
	tmp = (re * re) - (im * im);
end

Wolfram

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

Initial Program: 93.2% accurate, 1.0× speedup

Math

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

FPCore

(FPCore re_sqr (re im) :precision binary64 (- (* re re) (* im im)))

C

double re_sqr(double re, double im) {
	return (re * re) - (im * im);
}

Fortran

real(8) function re_sqr(re, im)
    real(8), intent (in) :: re
    real(8), intent (in) :: im
    re_sqr = (re * re) - (im * im)
end function

Java

public static double re_sqr(double re, double im) {
	return (re * re) - (im * im);
}

Python

def re_sqr(re, im):
	return (re * re) - (im * im)

Julia

function re_sqr(re, im)
	return Float64(Float64(re * re) - Float64(im * im))
end

MATLAB

function tmp = re_sqr(re, im)
	tmp = (re * re) - (im * im);
end

Wolfram

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

Alternative 1: 98.0% accurate, 0.1× speedup

Math

\[\begin{array}{l} re_m = \left|re\right| \\ \begin{array}{l} \mathbf{if}\;re\_m \leq 10^{+220}:\\ \;\;\;\;\mathsf{fma}\left(re\_m, re\_m, im \cdot \left(-im\right)\right)\\ \mathbf{else}:\\ \;\;\;\;\left(re\_m + im\right) \cdot \left(re\_m + im\right)\\ \end{array} \end{array} \]

FPCore

re_m = (fabs.f64 re)
(FPCore re_sqr (re_m im)
 :precision binary64
 (if (<= re_m 1e+220)
   (fma re_m re_m (* im (- im)))
   (* (+ re_m im) (+ re_m im))))

C

re_m = fabs(re);
double re_sqr(double re_m, double im) {
	double tmp;
	if (re_m <= 1e+220) {
		tmp = fma(re_m, re_m, (im * -im));
	} else {
		tmp = (re_m + im) * (re_m + im);
	}
	return tmp;
}

Julia

re_m = abs(re)
function re_sqr(re_m, im)
	tmp = 0.0
	if (re_m <= 1e+220)
		tmp = fma(re_m, re_m, Float64(im * Float64(-im)));
	else
		tmp = Float64(Float64(re_m + im) * Float64(re_m + im));
	end
	return tmp
end

Wolfram

re_m = N[Abs[re], $MachinePrecision]
re$95$sqr[re$95$m_, im_] := If[LessEqual[re$95$m, 1e+220], N[(re$95$m * re$95$m + N[(im * (-im)), $MachinePrecision]), $MachinePrecision], N[(N[(re$95$m + im), $MachinePrecision] * N[(re$95$m + im), $MachinePrecision]), $MachinePrecision]]

Derivation

1. Split input into 2 regimes

2. if re < 1e220

   1. Initial program 94.9%

      \[re \cdot re - im \cdot im \]
   2. Step-by-step derivation

      1. sqr-neg 94.9%

         \[\leadsto re \cdot re - \color{blue}{\left(-im\right) \cdot \left(-im\right)} \]

      2. cancel-sign-sub 94.9%

         \[\leadsto \color{blue}{re \cdot re + im \cdot \left(-im\right)} \]

      3. fma-define 96.6%

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

   3. Simplified 96.6%

      \[\leadsto \color{blue}{\mathsf{fma}\left(re, re, im \cdot \left(-im\right)\right)} \]
   4. Add Preprocessing

   if 1e220 < re

   1. Initial program 73.7%

      \[re \cdot re - im \cdot im \]
   2. Add Preprocessing
   3. Step-by-step derivation

      1. difference-of-squares 100.0%

         \[\leadsto \color{blue}{\left(re + im\right) \cdot \left(re - im\right)} \]

      2. sub-neg 100.0%

         \[\leadsto \left(re + im\right) \cdot \color{blue}{\left(re + \left(-im\right)\right)} \]

      3. add-sqr-sqrt 36.8%

         \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{-im} \cdot \sqrt{-im}}\right) \]

      4. sqrt-unprod 100.0%

         \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{\left(-im\right) \cdot \left(-im\right)}}\right) \]

      5. sqr-neg 100.0%

         \[\leadsto \left(re + im\right) \cdot \left(re + \sqrt{\color{blue}{im \cdot im}}\right) \]

      6. sqrt-prod 63.2%

         \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{im} \cdot \sqrt{im}}\right) \]

      7. add-sqr-sqrt 100.0%

         \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{im}\right) \]

   4. Applied egg-rr 100.0%

      \[\leadsto \color{blue}{\left(re + im\right) \cdot \left(re + im\right)} \]
3. Recombined 2 regimes into one program.

4. Final simplification 96.9%

   \[\leadsto \begin{array}{l} \mathbf{if}\;re \leq 10^{+220}:\\ \;\;\;\;\mathsf{fma}\left(re, re, im \cdot \left(-im\right)\right)\\ \mathbf{else}:\\ \;\;\;\;\left(re + im\right) \cdot \left(re + im\right)\\ \end{array} \]
5. Add Preprocessing

Alternative 2: 96.2% accurate, 0.6× speedup

Math

\[\begin{array}{l} re_m = \left|re\right| \\ \begin{array}{l} \mathbf{if}\;re\_m \leq 2.3 \cdot 10^{+150}:\\ \;\;\;\;re\_m \cdot re\_m - im \cdot im\\ \mathbf{else}:\\ \;\;\;\;\left(re\_m + im\right) \cdot \left(re\_m + im\right)\\ \end{array} \end{array} \]

FPCore

re_m = (fabs.f64 re)
(FPCore re_sqr (re_m im)
 :precision binary64
 (if (<= re_m 2.3e+150)
   (- (* re_m re_m) (* im im))
   (* (+ re_m im) (+ re_m im))))

C

re_m = fabs(re);
double re_sqr(double re_m, double im) {
	double tmp;
	if (re_m <= 2.3e+150) {
		tmp = (re_m * re_m) - (im * im);
	} else {
		tmp = (re_m + im) * (re_m + im);
	}
	return tmp;
}

Fortran

re_m = abs(re)
real(8) function re_sqr(re_m, im)
    real(8), intent (in) :: re_m
    real(8), intent (in) :: im
    real(8) :: tmp
    if (re_m <= 2.3d+150) then
        tmp = (re_m * re_m) - (im * im)
    else
        tmp = (re_m + im) * (re_m + im)
    end if
    re_sqr = tmp
end function

Java

re_m = Math.abs(re);
public static double re_sqr(double re_m, double im) {
	double tmp;
	if (re_m <= 2.3e+150) {
		tmp = (re_m * re_m) - (im * im);
	} else {
		tmp = (re_m + im) * (re_m + im);
	}
	return tmp;
}

Python

re_m = math.fabs(re)
def re_sqr(re_m, im):
	tmp = 0
	if re_m <= 2.3e+150:
		tmp = (re_m * re_m) - (im * im)
	else:
		tmp = (re_m + im) * (re_m + im)
	return tmp

Julia

re_m = abs(re)
function re_sqr(re_m, im)
	tmp = 0.0
	if (re_m <= 2.3e+150)
		tmp = Float64(Float64(re_m * re_m) - Float64(im * im));
	else
		tmp = Float64(Float64(re_m + im) * Float64(re_m + im));
	end
	return tmp
end

MATLAB

re_m = abs(re);
function tmp_2 = re_sqr(re_m, im)
	tmp = 0.0;
	if (re_m <= 2.3e+150)
		tmp = (re_m * re_m) - (im * im);
	else
		tmp = (re_m + im) * (re_m + im);
	end
	tmp_2 = tmp;
end

Wolfram

re_m = N[Abs[re], $MachinePrecision]
re$95$sqr[re$95$m_, im_] := If[LessEqual[re$95$m, 2.3e+150], N[(N[(re$95$m * re$95$m), $MachinePrecision] - N[(im * im), $MachinePrecision]), $MachinePrecision], N[(N[(re$95$m + im), $MachinePrecision] * N[(re$95$m + im), $MachinePrecision]), $MachinePrecision]]

Derivation

1. Split input into 2 regimes

2. if re < 2.30000000000000001e150

   1. Initial program 95.6%

      \[re \cdot re - im \cdot im \]
   2. Add Preprocessing

   if 2.30000000000000001e150 < re

   1. Initial program 77.4%

      \[re \cdot re - im \cdot im \]
   2. Add Preprocessing
   3. Step-by-step derivation

      1. difference-of-squares 100.0%

         \[\leadsto \color{blue}{\left(re + im\right) \cdot \left(re - im\right)} \]

      2. sub-neg 100.0%

         \[\leadsto \left(re + im\right) \cdot \color{blue}{\left(re + \left(-im\right)\right)} \]

      3. add-sqr-sqrt 51.6%

         \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{-im} \cdot \sqrt{-im}}\right) \]

      4. sqrt-unprod 100.0%

         \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{\left(-im\right) \cdot \left(-im\right)}}\right) \]

      5. sqr-neg 100.0%

         \[\leadsto \left(re + im\right) \cdot \left(re + \sqrt{\color{blue}{im \cdot im}}\right) \]

      6. sqrt-prod 48.4%

         \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{im} \cdot \sqrt{im}}\right) \]

      7. add-sqr-sqrt 93.5%

         \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{im}\right) \]

   4. Applied egg-rr 93.5%

      \[\leadsto \color{blue}{\left(re + im\right) \cdot \left(re + im\right)} \]
3. Recombined 2 regimes into one program.

4. Final simplification 95.3%

   \[\leadsto \begin{array}{l} \mathbf{if}\;re \leq 2.3 \cdot 10^{+150}:\\ \;\;\;\;re \cdot re - im \cdot im\\ \mathbf{else}:\\ \;\;\;\;\left(re + im\right) \cdot \left(re + im\right)\\ \end{array} \]
5. Add Preprocessing

Alternative 3: 52.8% accurate, 1.0× speedup

Math

\[\begin{array}{l} re_m = \left|re\right| \\ \left(re\_m + im\right) \cdot \left(re\_m + im\right) \end{array} \]

FPCore

re_m = (fabs.f64 re)
(FPCore re_sqr (re_m im) :precision binary64 (* (+ re_m im) (+ re_m im)))

C

re_m = fabs(re);
double re_sqr(double re_m, double im) {
	return (re_m + im) * (re_m + im);
}

Fortran

re_m = abs(re)
real(8) function re_sqr(re_m, im)
    real(8), intent (in) :: re_m
    real(8), intent (in) :: im
    re_sqr = (re_m + im) * (re_m + im)
end function

Java

re_m = Math.abs(re);
public static double re_sqr(double re_m, double im) {
	return (re_m + im) * (re_m + im);
}

Python

re_m = math.fabs(re)
def re_sqr(re_m, im):
	return (re_m + im) * (re_m + im)

Julia

re_m = abs(re)
function re_sqr(re_m, im)
	return Float64(Float64(re_m + im) * Float64(re_m + im))
end

MATLAB

re_m = abs(re);
function tmp = re_sqr(re_m, im)
	tmp = (re_m + im) * (re_m + im);
end

Wolfram

re_m = N[Abs[re], $MachinePrecision]
re$95$sqr[re$95$m_, im_] := N[(N[(re$95$m + im), $MachinePrecision] * N[(re$95$m + im), $MachinePrecision]), $MachinePrecision]

Derivation

1. Initial program 93.4%

   \[re \cdot re - im \cdot im \]
2. Add Preprocessing
3. Step-by-step derivation

   1. difference-of-squares 100.0%

      \[\leadsto \color{blue}{\left(re + im\right) \cdot \left(re - im\right)} \]

   2. sub-neg 100.0%

      \[\leadsto \left(re + im\right) \cdot \color{blue}{\left(re + \left(-im\right)\right)} \]

   3. add-sqr-sqrt 49.5%

      \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{-im} \cdot \sqrt{-im}}\right) \]

   4. sqrt-unprod 73.9%

      \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{\left(-im\right) \cdot \left(-im\right)}}\right) \]

   5. sqr-neg 73.9%

      \[\leadsto \left(re + im\right) \cdot \left(re + \sqrt{\color{blue}{im \cdot im}}\right) \]

   6. sqrt-prod 27.4%

      \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{\sqrt{im} \cdot \sqrt{im}}\right) \]

   7. add-sqr-sqrt 55.4%

      \[\leadsto \left(re + im\right) \cdot \left(re + \color{blue}{im}\right) \]

4. Applied egg-rr 55.4%

   \[\leadsto \color{blue}{\left(re + im\right) \cdot \left(re + im\right)} \]

5. Final simplification 55.4%

   \[\leadsto \left(re + im\right) \cdot \left(re + im\right) \]
6. Add Preprocessing

Reproduce

herbie shell --seed 2024078 
(FPCore re_sqr (re im)
  :name "math.square on complex, real part"
  :precision binary64
  (- (* re re) (* im im)))