diff options
Diffstat (limited to 'src/PackDepthwiseConvMatrixAvx2.cc')
-rw-r--r-- | src/PackDepthwiseConvMatrixAvx2.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/PackDepthwiseConvMatrixAvx2.cc b/src/PackDepthwiseConvMatrixAvx2.cc index ab2e1f2..04c08f3 100644 --- a/src/PackDepthwiseConvMatrixAvx2.cc +++ b/src/PackDepthwiseConvMatrixAvx2.cc @@ -205,7 +205,11 @@ void PackedDepthWiseConvMatrix::unpack(int8_t* unpacked_data) { } PackedDepthWiseConvMatrix::~PackedDepthWiseConvMatrix() { +#ifdef _MSC_VER + _aligned_free(pmat_); +#else free(pmat_); +#endif } } // namespace fbgemm |