nou/tenmon
1
0
Fork 1
Code Issues Pull Requests Projects Releases 33 Wiki Activity

Fix handling of inf and nan in TFloat16

Browse Source
This commit is contained in:
Dušan Poizl
2025-03-10 11:03:17 +01:00
parent bd45900821
commit d0dbef20c7
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tfloat16.h
+8 -1
View File
@@ -27,6 +27,12 @@ public:
b16 |= sign; b16 |= sign;
b16 |= mantisa; b16 |= mantisa;
} }
else if(exp == 255)//inf or nan
{
b16 = 0x7c00;
b16 |= sign;
b16 |= mantisa;
}
else if(exp > 142) else if(exp > 142)
{ {
b16 = 0x7c00;// inf b16 = 0x7c00;// inf
@@ -51,7 +57,8 @@ public:
if(b16) if(b16)
{ {
i |= sign << 16; i |= sign << 16;
i |= (exp + 112) << 23; if(exp==31)i |= 0x7f800000;
else i |= (exp + 112) << 23;
i |= (b16 & 0x3ff) << 13; i |= (b16 & 0x3ff) << 13;
} }
return *reinterpret_cast<float*>(&i); return *reinterpret_cast<float*>(&i);