Update QnnWrapperUtils.cpp

Signed-off-by: quic-zhanweiw <150764245+quic-zhanweiw@users.noreply.github.com>
quic · Sep 16, 2024 · 83fc01e · 83fc01e
1 parent af8f097
commit 83fc01e
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/WrapperUtils/QnnWrapperUtils.cpp b/src/WrapperUtils/QnnWrapperUtils.cpp
@@ -18,6 +18,13 @@ qnn_wrapper_api::ModelError_t qnn_wrapper_api::freeQnnTensor(Qnn_Tensor_t &tenso
   if (QNN_TENSOR_GET_IS_DYNAMIC_DIMENSIONS(tensor)) {
     free(QNN_TENSOR_GET_IS_DYNAMIC_DIMENSIONS(tensor));
   }
+  auto quant = QNN_TENSOR_GET_QUANT_PARAMS(tensor);
+  auto encoding = quant.quantizationEncoding;
+  if (encoding == QNN_QUANTIZATION_ENCODING_AXIS_SCALE_OFFSET) {
+      if (quant.axisScaleOffsetEncoding.scaleOffset != nullptr) {
+          free(quant.axisScaleOffsetEncoding.scaleOffset);
+      }
+  }
   return MODEL_NO_ERROR;
 }