BFPOWER/levlib
2
0
Fork 0
Code Issues 7 Pull Requests Actions Packages Projects Releases Wiki Activity

Switch from custom LN_2 constant to math.LN2

Browse Source
This commit is contained in:
Zachary Levy
2026-03-29 20:29:24 -07:00
parent a8f98e893f
commit 25fca052f5
2 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
.zed/tasks.json
View File

@@ -22,6 +22,11 @@
"command": "odin test levsync -out=out/debug/test_levsync", "command": "odin test levsync -out=out/debug/test_levsync",
"cwd": "$ZED_WORKTREE_ROOT" "cwd": "$ZED_WORKTREE_ROOT"
}, },
{
"label": "Test levmath",
"command": "odin test levmath -out=out/debug/test_levmath",
"cwd": "$ZED_WORKTREE_ROOT"
},
// --------------------------------------------------------------------------------------------------------------------- // ---------------------------------------------------------------------------------------------------------------------
// ----- LMDB Examples ------------------------ // ----- LMDB Examples ------------------------
// --------------------------------------------------------------------------------------------------------------------- // ---------------------------------------------------------------------------------------------------------------------

9
levmath/levmath.odin
View File

@@ -34,14 +34,13 @@ import "core:testing"
// Neural Computation 11, 853–862 (1999). // Neural Computation 11, 853–862 (1999).
// J. Rade, FastExp.h (2021), https://gist.github.com/jrade/293a73f89dfef51da6522428c857802d // J. Rade, FastExp.h (2021), https://gist.github.com/jrade/293a73f89dfef51da6522428c857802d
fast_exp :: #force_inline proc "contextless" (x: $FLOAT) -> FLOAT where intrinsics.type_is_float(FLOAT) { fast_exp :: #force_inline proc "contextless" (x: $FLOAT) -> FLOAT where intrinsics.type_is_float(FLOAT) {
LN_2 :: 0.6931471805599453
CORRECTION :: 0.04367744890362246 CORRECTION :: 0.04367744890362246
when FLOAT == f16 { when FLOAT == f16 {
MANTISSA_BITS :: 10 MANTISSA_BITS :: 10
EXPONENT_BIAS :: 15 EXPONENT_BIAS :: 15
SHIFT :: f16(1 << MANTISSA_BITS) SHIFT :: f16(1 << MANTISSA_BITS)
SCALE :: SHIFT / LN_2 SCALE :: SHIFT / math.LN2
BIAS :: SHIFT * f16(EXPONENT_BIAS - CORRECTION) BIAS :: SHIFT * f16(EXPONENT_BIAS - CORRECTION)
MAX_Y :: SHIFT * (2 * EXPONENT_BIAS + 1) MAX_Y :: SHIFT * (2 * EXPONENT_BIAS + 1)
@@ -52,7 +51,7 @@ fast_exp :: #force_inline proc "contextless" (x: $FLOAT) -> FLOAT where intrinsi
MANTISSA_BITS :: 23 MANTISSA_BITS :: 23
EXPONENT_BIAS :: 127 EXPONENT_BIAS :: 127
SHIFT :: f32(1 << MANTISSA_BITS) SHIFT :: f32(1 << MANTISSA_BITS)
SCALE :: SHIFT / LN_2 SCALE :: SHIFT / math.LN2
BIAS :: SHIFT * f32(EXPONENT_BIAS - CORRECTION) BIAS :: SHIFT * f32(EXPONENT_BIAS - CORRECTION)
MAX_Y :: SHIFT * (2 * EXPONENT_BIAS + 1) MAX_Y :: SHIFT * (2 * EXPONENT_BIAS + 1)
@@ -63,7 +62,7 @@ fast_exp :: #force_inline proc "contextless" (x: $FLOAT) -> FLOAT where intrinsi
MANTISSA_BITS :: 52 MANTISSA_BITS :: 52
EXPONENT_BIAS :: 1023 EXPONENT_BIAS :: 1023
SHIFT :: f64(1 << MANTISSA_BITS) SHIFT :: f64(1 << MANTISSA_BITS)
SCALE :: SHIFT / LN_2 SCALE :: SHIFT / math.LN2
BIAS :: SHIFT * f64(EXPONENT_BIAS - CORRECTION) BIAS :: SHIFT * f64(EXPONENT_BIAS - CORRECTION)
MAX_Y :: SHIFT * (2 * EXPONENT_BIAS + 1) MAX_Y :: SHIFT * (2 * EXPONENT_BIAS + 1)
@@ -174,4 +173,4 @@ test_fast_exp_saturation :: proc(t: ^testing.T) {
ov64 := fast_exp(f64(1e5)) ov64 := fast_exp(f64(1e5))
testing.expectf(t, math.is_inf_f32(ov32), "f32 overflow: got %v, want +inf", ov32) testing.expectf(t, math.is_inf_f32(ov32), "f32 overflow: got %v, want +inf", ov32)
testing.expectf(t, math.is_inf_f64(ov64), "f64 overflow: got %v, want +inf", ov64) testing.expectf(t, math.is_inf_f64(ov64), "f64 overflow: got %v, want +inf", ov64)
} }