Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/FBGEMM.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'include/fbgemm/Fbgemm.h')
-rw-r--r--include/fbgemm/Fbgemm.h12
1 files changed, 12 insertions, 0 deletions
diff --git a/include/fbgemm/Fbgemm.h b/include/fbgemm/Fbgemm.h
index 70f6294..09ddbe1 100644
--- a/include/fbgemm/Fbgemm.h
+++ b/include/fbgemm/Fbgemm.h
@@ -748,7 +748,11 @@ class FBGEMM_API PackAWithIm2Col
~PackAWithIm2Col() {
if (rowOffsetAllocatedHere) {
+#ifdef _MSC_VER
+ _aligned_free(row_offset_);
+#else
free(row_offset_);
+#endif
}
}
@@ -839,7 +843,11 @@ class FBGEMM_API PackAWithRowOffset final
~PackAWithRowOffset() {
if (rowOffsetAllocatedHere) {
+#ifdef _MSC_VER
+ _aligned_free(row_offset_);
+#else
free(row_offset_);
+#endif
}
}
@@ -932,7 +940,11 @@ class FBGEMM_API PackAWithQuantRowOffset final
~PackAWithQuantRowOffset() {
if (rowOffsetAllocatedHere) {
+#ifdef _MSC_VER
+ _aligned_free(row_offset_);
+#else
free(row_offset_);
+#endif
}
}