| ... | ... | @@ -10,12 +10,19 @@ const assert = std.debug.assert; |
| 10 | 10 | pub fn fabs(x: var) @typeOf(x) { |
| 11 | 11 | const T = @typeOf(x); |
| 12 | 12 | return switch (T) { |
| 13 | f16 => fabs16(x), |
| 13 | 14 | f32 => fabs32(x), |
| 14 | 15 | f64 => fabs64(x), |
| 15 | 16 | else => @compileError("fabs not implemented for " ++ @typeName(T)), |
| 16 | 17 | }; |
| 17 | 18 | } |
| 18 | 19 | |
| 20 | fn fabs16(x: f16) f16 { |
| 21 | var u = @bitCast(u16, x); |
| 22 | u &= 0x7FFF; |
| 23 | return @bitCast(f16, u); |
| 24 | } |
| 25 | |
| 19 | 26 | fn fabs32(x: f32) f32 { |
| 20 | 27 | var u = @bitCast(u32, x); |
| 21 | 28 | u &= 0x7FFFFFFF; |
| ... | ... | @@ -29,10 +36,16 @@ fn fabs64(x: f64) f64 { |
| 29 | 36 | } |
| 30 | 37 | |
| 31 | 38 | test "math.fabs" { |
| 39 | assert(fabs(f16(1.0)) == fabs16(1.0)); |
| 32 | 40 | assert(fabs(f32(1.0)) == fabs32(1.0)); |
| 33 | 41 | assert(fabs(f64(1.0)) == fabs64(1.0)); |
| 34 | 42 | } |
| 35 | 43 | |
| 44 | test "math.fabs16" { |
| 45 | assert(fabs16(1.0) == 1.0); |
| 46 | assert(fabs16(-1.0) == 1.0); |
| 47 | } |
| 48 | |
| 36 | 49 | test "math.fabs32" { |
| 37 | 50 | assert(fabs32(1.0) == 1.0); |
| 38 | 51 | assert(fabs32(-1.0) == 1.0); |
| ... | ... | @@ -43,6 +56,12 @@ test "math.fabs64" { |
| 43 | 56 | assert(fabs64(-1.0) == 1.0); |
| 44 | 57 | } |
| 45 | 58 | |
| 59 | test "math.fabs16.special" { |
| 60 | assert(math.isPositiveInf(fabs(math.inf(f16)))); |
| 61 | assert(math.isPositiveInf(fabs(-math.inf(f16)))); |
| 62 | assert(math.isNan(fabs(math.nan(f16)))); |
| 63 | } |
| 64 | |
| 46 | 65 | test "math.fabs32.special" { |
| 47 | 66 | assert(math.isPositiveInf(fabs(math.inf(f32)))); |
| 48 | 67 | assert(math.isPositiveInf(fabs(-math.inf(f32)))); |