Fix more MSVC inconsistent DLL linkage (#2064)

Summary: Pull Request resolved: #2064 Reviewed By: spcyppt Differential Revision: D49978436 Pulled By: q10 fbshipit-source-id: 74ec5bcc121b79774e20e14480a624e35f28f1e7
pytorch · Oct 6, 2023 · a684a79 · a684a79
1 parent c22d85c
commit a684a79
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 1 deletion.
diff --git a/src/EmbeddingSpMDMAvx2.cc b/src/EmbeddingSpMDMAvx2.cc
@@ -6,7 +6,7 @@
  * LICENSE file in the root directory of this source tree.
  */
 
-#include <cassert>
+#define FBGEMM_EXPORTS
 #include <cmath>
 #include "RefImplementations.h"
 #include "fbgemm/FbgemmEmbedding.h"

diff --git a/src/FbgemmBfloat16ConvertAvx2.cc b/src/FbgemmBfloat16ConvertAvx2.cc
@@ -10,6 +10,7 @@
     (defined(_MSC_VER) && (defined(_M_X64) || defined(_M_IX86)))
 #include <immintrin.h>
 #endif
+#define FBGEMM_EXPORTS
 #include "fbgemm/FbgemmConvert.h"
 
 namespace fbgemm {

diff --git a/src/FbgemmBfloat16ConvertAvx512.cc b/src/FbgemmBfloat16ConvertAvx512.cc
@@ -10,6 +10,7 @@
     (defined(_MSC_VER) && (defined(_M_X64) || defined(_M_IX86)))
 #include <immintrin.h>
 #endif
+#define FBGEMM_EXPORTS
 #include "fbgemm/FbgemmConvert.h"
 
 namespace fbgemm {

diff --git a/src/FbgemmFloat16ConvertAvx2.cc b/src/FbgemmFloat16ConvertAvx2.cc
@@ -10,6 +10,7 @@
     (defined(_MSC_VER) && (defined(_M_X64) || defined(_M_IX86)))
 #include <immintrin.h>
 #endif
+#define FBGEMM_EXPORTS
 #include "fbgemm/FbgemmConvert.h"
 
 namespace fbgemm {

diff --git a/src/FbgemmFloat16ConvertAvx512.cc b/src/FbgemmFloat16ConvertAvx512.cc
@@ -10,6 +10,7 @@
     (defined(_MSC_VER) && (defined(_M_X64) || defined(_M_IX86)))
 #include <immintrin.h>
 #endif
+#define FBGEMM_EXPORTS
 #include "fbgemm/FbgemmConvert.h"
 
 namespace fbgemm {