commit 60dc48c4ddf30f2a76d4cfcf1b40ca57b6f3bf95
author Tai Ly <tai.ly@arm.com> Fri Mar 08 22:19:41 2024 +0000
committer Tai Ly <tai.ly@arm.com> Sun Mar 17 19:56:21 2024 -0700
tree e3d229a2d596e1a0788dfd75d77b996263055496
parent e67115ef82bcba0718dcbd75cc8411985001b7cc

[ref model] Change Clamp and Pad attribute fields

This implements changes due to ClampAttribute and PadAttribute
field changes.

Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Ide01e2a27fe3c1ea7794e7a4b6780b7eae436caf

reference_model/src/ops/data_layout.cc

diff --git a/reference_model/src/ops/data_layout.cc b/reference_model/src/ops/data_layout.cc
index b6ad704..4c17e78 100644
--- a/reference_model/src/ops/data_layout.cc
+++ b/reference_model/src/ops/data_layout.cc
@@ -176,17 +176,25 @@
         case TOSA_REF_TYPE_BOOL:
         case TOSA_REF_TYPE_INT8:
         case TOSA_REF_TYPE_INT16:
-        case TOSA_REF_TYPE_INT32:
-            pad_value = (InEigenType)attribute->pad_const_int();
+        case TOSA_REF_TYPE_INT32: {
+            std::vector<int32_t> int32_data;
+            TosaSerializationHandler::ConvertU8toI32(attribute->pad_const(),
+                                                     /* size = */ 1, int32_data);
+            pad_value = (InEigenType)int32_data[0];
             break;
+        }
         case TOSA_REF_TYPE_FP16:
         case TOSA_REF_TYPE_BF16:
         case TOSA_REF_TYPE_FP32:
         case TOSA_REF_TYPE_FP64:
         case TOSA_REF_TYPE_FP8E4M3:
-        case TOSA_REF_TYPE_FP8E5M2:
-            pad_value = (InEigenType)attribute->pad_const_fp();
+        case TOSA_REF_TYPE_FP8E5M2: {
+            std::vector<float> float_data;
+            TosaSerializationHandler::ConvertU8toF32(attribute->pad_const(),
+                                                     /* size = */ 1, float_data);
+            pad_value = (InEigenType)float_data[0];
             break;
+        }
         default:
             ASSERT_MSG(false, "TOSA_REF_TYPE %s is not supported.", EnumNameTOSAREFTYPE(Dtype));
             break;