diff options
Diffstat (limited to 'include/fbgemm/Fbgemm.h')
-rw-r--r-- | include/fbgemm/Fbgemm.h | 12 |
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 } } |