From 1acb3672107eeb94c7c23d13c24df3d7671dbcc6 Mon Sep 17 00:00:00 2001
From: Tai Ly <tai.ly@arm.com>
Date: Fri, 14 Jul 2023 17:02:12 +0000
Subject: Fix PadOp serialization for input_zp

This fixes previously introduced bug where the input_zp was
dropped from the PadOp serialization.

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I42fec4a2cdad0ab1ea3cc2f188c4f4c4467583b0
---
 src/TosaSerialize.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'src/TosaSerialize.cpp')

diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index fec9f17..66d0a31 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -931,7 +931,9 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::PadOp>(
     paddings.push_back(val);
   }
 
-  int32_t pad_const_int = 0;
+  auto quant_info = pad_op.getQuantizationInfoAttr();
+  // pad_const includes the zero point if the tensor uses a zero point.
+  int32_t pad_const_int = quant_info ? quant_info.getInputZp() : 0;
   float pad_const_fp = 0.f;
 
   if (auto tensor = pad_op.getPadConst()) {
-- 
cgit v1.2.1