Fix compilaton of bfloat16 on Windows (#4415)

34c636d9 · Philip Hyunsu Cho · Tianqi Chen · 2f1685fe · 34c636d9
Commit 34c636d9 authored Nov 24, 2019 by Philip Hyunsu Cho Committed by Tianqi Chen Nov 24, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 3 deletions

3rdparty/bfloat16/bfloat16.cc
+7 -3

No files found.
--- a/3rdparty/bfloat16/bfloat16.cc
+++ b/3rdparty/bfloat16/bfloat16.cc
@@ -60,19 +60,23 @@ void BFloat16Add(const uint16_t* a, const uint16_t* b, uint16_t* dst,
 }

 extern "C" {
-TVM_DLL TVM_DLL uint16_t FloatToBFloat16_wrapper(float in) {
+TVM_DLL uint16_t FloatToBFloat16_wrapper(float in);
+TVM_DLL float BFloat16ToFloat_wrapper(uint16_t in);
+TVM_DLL uint16_t BFloat16Add_wrapper(uint16_t a, uint16_t b);
+
+uint16_t FloatToBFloat16_wrapper(float in) {
  uint16_t out;
  FloatToBFloat16(&in, &out, 1);
  return out;
 }

-TVM_DLL float BFloat16ToFloat_wrapper(uint16_t in) {
+float BFloat16ToFloat_wrapper(uint16_t in) {
  float out;
  BFloat16ToFloat(&in, &out, 1);
  return out;
 }

-TVM_DLL uint16_t BFloat16Add_wrapper(uint16_t a, uint16_t b) {
+uint16_t BFloat16Add_wrapper(uint16_t a, uint16_t b) {
  uint16_t out;
  BFloat16Add(&a, &b, &out, 1);
  return out;