export builtin_fp16 on Windows (#4731)

4e1ca857 · vexilligera · Tianqi Chen · 3279957f · 4e1ca857
Commit 4e1ca857 authored Jan 17, 2020 by vexilligera Committed by Tianqi Chen Jan 17, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 0 deletions

src/runtime/builtin_fp16.cc
+10 -0

No files found.
--- a/src/runtime/builtin_fp16.cc
+++ b/src/runtime/builtin_fp16.cc
@@ -37,5 +37,15 @@ TVM_DLL TVM_WEAK float __gnu_h2f_ieee(uint16_t a) {
  return __extendXfYf2__<uint16_t, uint16_t, 10, float, uint32_t, 23>(a);
 }

+#else
+
+TVM_DLL uint16_t __gnu_f2h_ieee(float a) {
+  return __truncXfYf2__<float, uint32_t, 23, uint16_t, uint16_t, 10>(a);
+}
+
+TVM_DLL float __gnu_h2f_ieee(uint16_t a) {
+  return __extendXfYf2__<uint16_t, uint16_t, 10, float, uint32_t, 23>(a);
+}
+
 #endif
 }