From 7feb33758e5662d328a23add6c914e53142b0b33 Mon Sep 17 00:00:00 2001
From: Young Jin Kim <youki@microsoft.com>
Date: Thu, 21 May 2020 14:44:33 -0700
Subject: Remove an unnecessary memory allocation

---
 src/PackAWithQuantRowOffset.cc | 6 ++++--
 1 file 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>
-- 
cgit v1.2.3