authorgravatar for info@bnoordhuis.nlBen Noordhuis <info@bnoordhuis.nl> 2018-06-30 01:44:54+02:00
committergravatar for info@bnoordhuis.nlBen Noordhuis <info@bnoordhuis.nl> 2018-06-30 01:58:17+02:00
log27b02413dc3dacc7d784fe84ff8ba6cb0361842d
treea48f969278184b262aaae0005583bc182bea45b6
parent61df5bc142253b0d33b77bceecfaef4c767e7feb

add std.math f16 nan support

refs #1122

5 files changed, 15 insertions(+), 1 deletions(-)

std/math/index.zig+3
......@@ -25,6 +25,9 @@ pub const f16_max = 65504;
2525pub const f16_epsilon = 0.0009765625; // 2**-10
2626pub const f16_toint = 1.0 / f16_epsilon;
2727
28pub const nan_u16 = u16(0x7C01);
29pub const nan_f16 = @bitCast(f16, nan_u16);
30
2831pub const nan_u32 = u32(0x7F800001);
2932pub const nan_f32 = @bitCast(f32, nan_u32);
3033
std/math/isnan.zig+6
......@@ -5,6 +5,10 @@ const assert = std.debug.assert;
55pub fn isNan(x: var) bool {
66 const T = @typeOf(x);
77 switch (T) {
8 f16 => {
9 const bits = @bitCast(u16, x);
10 return (bits & 0x7fff) > 0x7c00;
11 },
812 f32 => {
913 const bits = @bitCast(u32, x);
1014 return bits & 0x7FFFFFFF > 0x7F800000;
......@@ -26,8 +30,10 @@ pub fn isSignalNan(x: var) bool {
2630}
2731
2832test "math.isNan" {
33 assert(isNan(math.nan(f16)));
2934 assert(isNan(math.nan(f32)));
3035 assert(isNan(math.nan(f64)));
36 assert(!isNan(f16(1.0)));
3137 assert(!isNan(f32(1.0)));
3238 assert(!isNan(f64(1.0)));
3339}
std/math/nan.zig+2
......@@ -2,6 +2,7 @@ const math = @import("index.zig");
22
33pub fn nan(comptime T: type) T {
44 return switch (T) {
5 f16 => @bitCast(f16, math.nan_u16),
56 f32 => @bitCast(f32, math.nan_u32),
67 f64 => @bitCast(f64, math.nan_u64),
78 else => @compileError("nan not implemented for " ++ @typeName(T)),
......@@ -12,6 +13,7 @@ pub fn nan(comptime T: type) T {
1213// representation in the future when required.
1314pub fn snan(comptime T: type) T {
1415 return switch (T) {
16 f16 => @bitCast(f16, math.nan_u16),
1517 f32 => @bitCast(f32, math.nan_u32),
1618 f64 => @bitCast(f64, math.nan_u64),
1719 else => @compileError("snan not implemented for " ++ @typeName(T)),
std/special/builtin.zig+3-1
......@@ -210,7 +210,9 @@ fn generic_fmod(comptime T: type, x: T, y: T) T {
210210}
211211
212212fn isNan(comptime T: type, bits: T) bool {
213 if (T == u32) {
213 if (T == u16) {
214 return (bits & 0x7fff) > 0x7c00;
215 } else if (T == u32) {
214216 return (bits & 0x7fffffff) > 0x7f800000;
215217 } else if (T == u64) {
216218 return (bits & (@maxValue(u64) >> 1)) > (u64(0x7ff) << 52);
std/special/compiler_rt/extendXfYf2_test.zig+1
......@@ -88,6 +88,7 @@ test "extenddftf2" {
8888test "extendhfsf2" {
8989 test__extendhfsf2(0x7e00, 0x7fc00000); // qNaN
9090 test__extendhfsf2(0x7f00, 0x7fe00000); // sNaN
91 test__extendhfsf2(0x7c01, 0x7f802000); // sNaN
9192
9293 test__extendhfsf2(0, 0); // 0
9394 test__extendhfsf2(0x8000, 0x80000000); // -0