
(FPCore (u s)
:precision binary32
(let* ((t_0 (/ 1.0 (+ 1.0 (exp (/ PI s))))))
(*
(- s)
(log
(-
(/ 1.0 (+ (* u (- (/ 1.0 (+ 1.0 (exp (/ (- PI) s)))) t_0)) t_0))
1.0)))))
float code(float u, float s) {
float t_0 = 1.0f / (1.0f + expf((((float) M_PI) / s)));
return -s * logf(((1.0f / ((u * ((1.0f / (1.0f + expf((-((float) M_PI) / s)))) - t_0)) + t_0)) - 1.0f));
}
function code(u, s) t_0 = Float32(Float32(1.0) / Float32(Float32(1.0) + exp(Float32(Float32(pi) / s)))) return Float32(Float32(-s) * log(Float32(Float32(Float32(1.0) / Float32(Float32(u * Float32(Float32(Float32(1.0) / Float32(Float32(1.0) + exp(Float32(Float32(-Float32(pi)) / s)))) - t_0)) + t_0)) - Float32(1.0)))) end
function tmp = code(u, s) t_0 = single(1.0) / (single(1.0) + exp((single(pi) / s))); tmp = -s * log(((single(1.0) / ((u * ((single(1.0) / (single(1.0) + exp((-single(pi) / s)))) - t_0)) + t_0)) - single(1.0))); end
\begin{array}{l}
\\
\begin{array}{l}
t_0 := \frac{1}{1 + e^{\frac{\pi}{s}}}\\
\left(-s\right) \cdot \log \left(\frac{1}{u \cdot \left(\frac{1}{1 + e^{\frac{-\pi}{s}}} - t_0\right) + t_0} - 1\right)
\end{array}
\end{array}
Sampling outcomes in binary32 precision:
Herbie found 7 alternatives:
| Alternative | Accuracy | Speedup |
|---|
(FPCore (u s)
:precision binary32
(let* ((t_0 (/ 1.0 (+ 1.0 (exp (/ PI s))))))
(*
(- s)
(log
(-
(/ 1.0 (+ (* u (- (/ 1.0 (+ 1.0 (exp (/ (- PI) s)))) t_0)) t_0))
1.0)))))
float code(float u, float s) {
float t_0 = 1.0f / (1.0f + expf((((float) M_PI) / s)));
return -s * logf(((1.0f / ((u * ((1.0f / (1.0f + expf((-((float) M_PI) / s)))) - t_0)) + t_0)) - 1.0f));
}
function code(u, s) t_0 = Float32(Float32(1.0) / Float32(Float32(1.0) + exp(Float32(Float32(pi) / s)))) return Float32(Float32(-s) * log(Float32(Float32(Float32(1.0) / Float32(Float32(u * Float32(Float32(Float32(1.0) / Float32(Float32(1.0) + exp(Float32(Float32(-Float32(pi)) / s)))) - t_0)) + t_0)) - Float32(1.0)))) end
function tmp = code(u, s) t_0 = single(1.0) / (single(1.0) + exp((single(pi) / s))); tmp = -s * log(((single(1.0) / ((u * ((single(1.0) / (single(1.0) + exp((-single(pi) / s)))) - t_0)) + t_0)) - single(1.0))); end
\begin{array}{l}
\\
\begin{array}{l}
t_0 := \frac{1}{1 + e^{\frac{\pi}{s}}}\\
\left(-s\right) \cdot \log \left(\frac{1}{u \cdot \left(\frac{1}{1 + e^{\frac{-\pi}{s}}} - t_0\right) + t_0} - 1\right)
\end{array}
\end{array}
(FPCore (u s)
:precision binary32
(*
s
(-
(log
(+
(/
1.0
(+
(/ u (+ 1.0 (exp (/ (- PI) s))))
(/ (- 1.0 u) (+ 1.0 (exp (/ PI s))))))
-1.0)))))
float code(float u, float s) {
return s * -logf(((1.0f / ((u / (1.0f + expf((-((float) M_PI) / s)))) + ((1.0f - u) / (1.0f + expf((((float) M_PI) / s)))))) + -1.0f));
}
function code(u, s) return Float32(s * Float32(-log(Float32(Float32(Float32(1.0) / Float32(Float32(u / Float32(Float32(1.0) + exp(Float32(Float32(-Float32(pi)) / s)))) + Float32(Float32(Float32(1.0) - u) / Float32(Float32(1.0) + exp(Float32(Float32(pi) / s)))))) + Float32(-1.0))))) end
function tmp = code(u, s) tmp = s * -log(((single(1.0) / ((u / (single(1.0) + exp((-single(pi) / s)))) + ((single(1.0) - u) / (single(1.0) + exp((single(pi) / s)))))) + single(-1.0))); end
\begin{array}{l}
\\
s \cdot \left(-\log \left(\frac{1}{\frac{u}{1 + e^{\frac{-\pi}{s}}} + \frac{1 - u}{1 + e^{\frac{\pi}{s}}}} + -1\right)\right)
\end{array}
Initial program 98.9%
distribute-lft-neg-out98.9%
distribute-rgt-neg-in98.9%
sub-neg98.9%
Simplified98.9%
Final simplification98.9%
(FPCore (u s) :precision binary32 (- (/ (* s (- s)) PI) (* s (fma -1.0 (log s) (log PI)))))
float code(float u, float s) {
return ((s * -s) / ((float) M_PI)) - (s * fmaf(-1.0f, logf(s), logf(((float) M_PI))));
}
function code(u, s) return Float32(Float32(Float32(s * Float32(-s)) / Float32(pi)) - Float32(s * fma(Float32(-1.0), log(s), log(Float32(pi))))) end
\begin{array}{l}
\\
\frac{s \cdot \left(-s\right)}{\pi} - s \cdot \mathsf{fma}\left(-1, \log s, \log \pi\right)
\end{array}
Initial program 98.9%
distribute-lft-neg-out98.9%
distribute-rgt-neg-in98.9%
sub-neg98.9%
Simplified98.9%
Taylor expanded in s around -inf 24.7%
+-commutative24.7%
fma-def24.7%
associate--r+24.7%
cancel-sign-sub-inv24.7%
distribute-rgt-out--24.7%
*-commutative24.7%
metadata-eval24.7%
metadata-eval24.7%
*-commutative24.7%
Simplified24.7%
Taylor expanded in u around 0 24.9%
*-commutative24.9%
Simplified24.9%
Taylor expanded in s around 0 25.0%
distribute-lft-out25.0%
unpow225.0%
fma-def25.0%
Simplified25.0%
Final simplification25.0%
(FPCore (u s) :precision binary32 (* s (- (log s) (+ (log PI) (/ s PI)))))
float code(float u, float s) {
return s * (logf(s) - (logf(((float) M_PI)) + (s / ((float) M_PI))));
}
function code(u, s) return Float32(s * Float32(log(s) - Float32(log(Float32(pi)) + Float32(s / Float32(pi))))) end
function tmp = code(u, s) tmp = s * (log(s) - (log(single(pi)) + (s / single(pi)))); end
\begin{array}{l}
\\
s \cdot \left(\log s - \left(\log \pi + \frac{s}{\pi}\right)\right)
\end{array}
Initial program 98.9%
distribute-lft-neg-out98.9%
distribute-rgt-neg-in98.9%
sub-neg98.9%
Simplified98.9%
Taylor expanded in s around -inf 24.7%
+-commutative24.7%
fma-def24.7%
associate--r+24.7%
cancel-sign-sub-inv24.7%
distribute-rgt-out--24.7%
*-commutative24.7%
metadata-eval24.7%
metadata-eval24.7%
*-commutative24.7%
Simplified24.7%
Taylor expanded in u around 0 24.9%
*-commutative24.9%
Simplified24.9%
Taylor expanded in s around 0 25.0%
Final simplification25.0%
(FPCore (u s) :precision binary32 (* s (log (/ s PI))))
float code(float u, float s) {
return s * logf((s / ((float) M_PI)));
}
function code(u, s) return Float32(s * log(Float32(s / Float32(pi)))) end
function tmp = code(u, s) tmp = s * log((s / single(pi))); end
\begin{array}{l}
\\
s \cdot \log \left(\frac{s}{\pi}\right)
\end{array}
Initial program 98.9%
distribute-lft-neg-out98.9%
distribute-rgt-neg-in98.9%
sub-neg98.9%
Simplified98.9%
Taylor expanded in s around -inf 24.7%
+-commutative24.7%
fma-def24.7%
associate--r+24.7%
cancel-sign-sub-inv24.7%
distribute-rgt-out--24.7%
*-commutative24.7%
metadata-eval24.7%
metadata-eval24.7%
*-commutative24.7%
Simplified24.7%
Taylor expanded in u around 0 24.9%
*-commutative24.9%
Simplified24.9%
Taylor expanded in s around 0 25.0%
Taylor expanded in s around 0 25.0%
mul-1-neg25.0%
neg-mul-125.0%
+-commutative25.0%
sub-neg25.0%
log-div24.9%
distribute-rgt-neg-in24.9%
log-div25.0%
sub-neg25.0%
+-commutative25.0%
distribute-neg-in25.0%
remove-double-neg25.0%
sub-neg25.0%
log-div25.0%
Simplified25.0%
Final simplification25.0%
(FPCore (u s) :precision binary32 (if (<= s 9.999999682655225e-21) (* s 0.0) (* 4.0 (* PI (+ (* u 0.5) -0.25)))))
float code(float u, float s) {
float tmp;
if (s <= 9.999999682655225e-21f) {
tmp = s * 0.0f;
} else {
tmp = 4.0f * (((float) M_PI) * ((u * 0.5f) + -0.25f));
}
return tmp;
}
function code(u, s) tmp = Float32(0.0) if (s <= Float32(9.999999682655225e-21)) tmp = Float32(s * Float32(0.0)); else tmp = Float32(Float32(4.0) * Float32(Float32(pi) * Float32(Float32(u * Float32(0.5)) + Float32(-0.25)))); end return tmp end
function tmp_2 = code(u, s) tmp = single(0.0); if (s <= single(9.999999682655225e-21)) tmp = s * single(0.0); else tmp = single(4.0) * (single(pi) * ((u * single(0.5)) + single(-0.25))); end tmp_2 = tmp; end
\begin{array}{l}
\\
\begin{array}{l}
\mathbf{if}\;s \leq 9.999999682655225 \cdot 10^{-21}:\\
\;\;\;\;s \cdot 0\\
\mathbf{else}:\\
\;\;\;\;4 \cdot \left(\pi \cdot \left(u \cdot 0.5 + -0.25\right)\right)\\
\end{array}
\end{array}
if s < 9.99999968e-21Initial program 99.0%
distribute-lft-neg-out99.0%
distribute-rgt-neg-in99.0%
sub-neg99.0%
Simplified99.0%
Taylor expanded in s around inf 14.0%
if 9.99999968e-21 < s Initial program 98.8%
distribute-lft-neg-out98.8%
distribute-rgt-neg-in98.8%
sub-neg98.8%
Simplified98.8%
Taylor expanded in s around inf 14.3%
associate--r+14.3%
cancel-sign-sub-inv14.3%
distribute-rgt-out--14.3%
*-commutative14.3%
metadata-eval14.3%
metadata-eval14.3%
*-commutative14.3%
Simplified14.3%
add-cube-cbrt14.3%
pow314.3%
Applied egg-rr14.3%
Taylor expanded in u around 0 14.3%
+-commutative14.3%
*-commutative14.3%
*-commutative14.3%
*-commutative14.3%
associate-*r*14.3%
distribute-lft-out14.3%
*-commutative14.3%
Simplified14.3%
Final simplification14.1%
(FPCore (u s) :precision binary32 (- PI))
float code(float u, float s) {
return -((float) M_PI);
}
function code(u, s) return Float32(-Float32(pi)) end
function tmp = code(u, s) tmp = -single(pi); end
\begin{array}{l}
\\
-\pi
\end{array}
Initial program 98.9%
distribute-lft-neg-out98.9%
distribute-rgt-neg-in98.9%
sub-neg98.9%
Simplified98.9%
Taylor expanded in u around 0 10.5%
mul-1-neg10.5%
Simplified10.5%
Final simplification10.5%
(FPCore (u s) :precision binary32 (* s 0.0))
float code(float u, float s) {
return s * 0.0f;
}
real(4) function code(u, s)
real(4), intent (in) :: u
real(4), intent (in) :: s
code = s * 0.0e0
end function
function code(u, s) return Float32(s * Float32(0.0)) end
function tmp = code(u, s) tmp = s * single(0.0); end
\begin{array}{l}
\\
s \cdot 0
\end{array}
Initial program 98.9%
distribute-lft-neg-out98.9%
distribute-rgt-neg-in98.9%
sub-neg98.9%
Simplified98.9%
Taylor expanded in s around inf 11.0%
Final simplification11.0%
herbie shell --seed 2023229
(FPCore (u s)
:name "Sample trimmed logistic on [-pi, pi]"
:precision binary32
:pre (and (and (<= 2.328306437e-10 u) (<= u 1.0)) (and (<= 0.0 s) (<= s 1.0651631)))
(* (- s) (log (- (/ 1.0 (+ (* u (- (/ 1.0 (+ 1.0 (exp (/ (- PI) s)))) (/ 1.0 (+ 1.0 (exp (/ PI s)))))) (/ 1.0 (+ 1.0 (exp (/ PI s)))))) 1.0))))