bug366, discussion (missed optimization)

Average Accuracy: 50.9% → 98.9%

Time: 5.2s

Precision: binary64

Cost: 7044

Math

\[\sqrt{a \cdot a - b \cdot b} \]

↓

\[\begin{array}{l} \mathbf{if}\;a \leq -1 \cdot 10^{-238}:\\ \;\;\;\;\mathsf{fma}\left(0.5, \frac{b}{\frac{a}{b}}, -a\right)\\ \mathbf{else}:\\ \;\;\;\;a + -0.5 \cdot \left(b \cdot \frac{b}{a}\right)\\ \end{array} \]

FPCore

(FPCore (a b) :precision binary64 (sqrt (- (* a a) (* b b))))

↓

(FPCore (a b)
 :precision binary64
 (if (<= a -1e-238)
   (fma 0.5 (/ b (/ a b)) (- a))
   (+ a (* -0.5 (* b (/ b a))))))

C

double code(double a, double b) {
	return sqrt(((a * a) - (b * b)));
}

↓

double code(double a, double b) {
	double tmp;
	if (a <= -1e-238) {
		tmp = fma(0.5, (b / (a / b)), -a);
	} else {
		tmp = a + (-0.5 * (b * (b / a)));
	}
	return tmp;
}

Julia

function code(a, b)
	return sqrt(Float64(Float64(a * a) - Float64(b * b)))
end

↓

function code(a, b)
	tmp = 0.0
	if (a <= -1e-238)
		tmp = fma(0.5, Float64(b / Float64(a / b)), Float64(-a));
	else
		tmp = Float64(a + Float64(-0.5 * Float64(b * Float64(b / a))));
	end
	return tmp
end

Wolfram

code[a_, b_] := N[Sqrt[N[(N[(a * a), $MachinePrecision] - N[(b * b), $MachinePrecision]), $MachinePrecision]], $MachinePrecision]

↓

code[a_, b_] := If[LessEqual[a, -1e-238], N[(0.5 * N[(b / N[(a / b), $MachinePrecision]), $MachinePrecision] + (-a)), $MachinePrecision], N[(a + N[(-0.5 * N[(b * N[(b / a), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]), $MachinePrecision]]

Target

Original	50.9%
Target	99.2%
Herbie	98.9%

\[\sqrt{\left|a\right| + \left|b\right|} \cdot \sqrt{\left|a\right| - \left|b\right|} \]

Derivation

1. Split input into 2 regimes

2. if a < -9.9999999999999999e-239

   1. Initial program 51.8%

      \[\sqrt{a \cdot a - b \cdot b} \]

   2. Taylor expanded in a around -inf 92.6%

      \[\leadsto \color{blue}{0.5 \cdot \frac{{b}^{2}}{a} + -1 \cdot a} \]

   3. Simplified 99.4%

      \[\leadsto \color{blue}{\mathsf{fma}\left(0.5, \frac{b}{\frac{a}{b}}, -a\right)} \]
      Proof

      [Start] 92.6	\[ 0.5 \cdot \frac{{b}^{2}}{a} + -1 \cdot a \]
      fma-def [=>] 92.6	\[ \color{blue}{\mathsf{fma}\left(0.5, \frac{{b}^{2}}{a}, -1 \cdot a\right)} \]
      unpow2 [=>] 92.6	\[ \mathsf{fma}\left(0.5, \frac{\color{blue}{b \cdot b}}{a}, -1 \cdot a\right) \]
      associate-/l* [=>] 99.4	\[ \mathsf{fma}\left(0.5, \color{blue}{\frac{b}{\frac{a}{b}}}, -1 \cdot a\right) \]
      mul-1-neg [=>] 99.4	\[ \mathsf{fma}\left(0.5, \frac{b}{\frac{a}{b}}, \color{blue}{-a}\right) \]

   if -9.9999999999999999e-239 < a

   1. Initial program 50.1%

      \[\sqrt{a \cdot a - b \cdot b} \]

   2. Taylor expanded in a around inf 91.8%

      \[\leadsto \color{blue}{a + -0.5 \cdot \frac{{b}^{2}}{a}} \]

   3. Simplified 91.8%

      \[\leadsto \color{blue}{a + -0.5 \cdot \frac{b \cdot b}{a}} \]
      Proof

      [Start] 91.8	\[ a + -0.5 \cdot \frac{{b}^{2}}{a} \]
      unpow2 [=>] 91.8	\[ a + -0.5 \cdot \frac{\color{blue}{b \cdot b}}{a} \]

   4. Applied egg-rr 98.4%

      \[\leadsto a + -0.5 \cdot \color{blue}{\left(\frac{b}{a} \cdot b\right)} \]
      Proof

      [Start] 91.8	\[ a + -0.5 \cdot \frac{b \cdot b}{a} \]
      associate-/l* [=>] 98.4	\[ a + -0.5 \cdot \color{blue}{\frac{b}{\frac{a}{b}}} \]
      associate-/r/ [=>] 98.4	\[ a + -0.5 \cdot \color{blue}{\left(\frac{b}{a} \cdot b\right)} \]

3. Recombined 2 regimes into one program.

4. Final simplification 98.9%

   \[\leadsto \begin{array}{l} \mathbf{if}\;a \leq -1 \cdot 10^{-238}:\\ \;\;\;\;\mathsf{fma}\left(0.5, \frac{b}{\frac{a}{b}}, -a\right)\\ \mathbf{else}:\\ \;\;\;\;a + -0.5 \cdot \left(b \cdot \frac{b}{a}\right)\\ \end{array} \]

Alternatives

Alternative 1
Accuracy	99.1%
Cost	708

\[\begin{array}{l} \mathbf{if}\;a \leq -4 \cdot 10^{-302}:\\ \;\;\;\;-a\\ \mathbf{else}:\\ \;\;\;\;a + -0.5 \cdot \left(b \cdot \frac{b}{a}\right)\\ \end{array} \]

Alternative 2
Accuracy	98.3%
Cost	260

\[\begin{array}{l} \mathbf{if}\;a \leq -1 \cdot 10^{-238}:\\ \;\;\;\;-a\\ \mathbf{else}:\\ \;\;\;\;a\\ \end{array} \]

Alternative 3
Accuracy	51.1%
Cost	64

\[a \]

Reproduce

herbie shell --seed 2023131 
(FPCore (a b)
  :name "bug366, discussion (missed optimization)"
  :precision binary64

  :herbie-target
  (* (sqrt (+ (fabs a) (fabs b))) (sqrt (- (fabs a) (fabs b))))

  (sqrt (- (* a a) (* b b))))