Fix compilaton of bfloat16 on Windows (#4415)

apache · Nov 24, 2019 · 34c636d · 34c636d
1 parent 2f1685f
commit 34c636d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git 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;