diff options
| author | Young Jin Kim <youki@microsoft.com> | 2020-05-22 00:44:33 +0300 |
|---|---|---|
| committer | Young Jin Kim <youki@microsoft.com> | 2020-05-22 00:44:33 +0300 |
| commit | 7feb33758e5662d328a23add6c914e53142b0b33 (patch) | |
| tree | c00c8e6063128b6bf6e72a266042293b56360a54 | |
| parent | f78e60988329b9207d086c743cafce1ac1bea3ab (diff) | |
Remove an unnecessary memory allocationyouki/improve-mem-alloc-marian
| -rw-r--r-- | src/PackAWithQuantRowOffset.cc | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/PackAWithQuantRowOffset.cc b/src/PackAWithQuantRowOffset.cc index 13a8fad..31dec05 100644 --- a/src/PackAWithQuantRowOffset.cc +++ b/src/PackAWithQuantRowOffset.cc @@ -116,8 +116,9 @@ void PackAWithQuantRowOffset<T, accT>::pack(const block_type_t& block) { (block.col_start % (this->numCols() / this->numGroups())) != 0; int32_t* row_offset_buf = getRowOffsetBuffer(); - float* smat_transposed = new float[block.row_size * block.col_size]; + float* smat_transposed = nullptr; if (tr) { + smat_transposed = new float[block.row_size * block.col_size]; transpose_simd( block.col_size, block.row_size, @@ -156,7 +157,8 @@ void PackAWithQuantRowOffset<T, accT>::pack(const block_type_t& block) { } } - delete[] smat_transposed; + if (smat_transposed) + delete[] smat_transposed; } template <typename T, typename accT> |