commit 3195a665e3f96809a67b4cb04a57330d2bfeb0de
author Won Jeon <won.jeon@arm.com> Tue Feb 27 17:52:45 2024 +0000
committer Won Jeon <won.jeon@arm.com> Thu Feb 29 05:15:59 2024 +0000
tree 6ebcb1b8dbbfc9b992ac2c88095b892c8356f1e6
parent 687805380e3b5aafb48ee1eb2b29857cc869f880

Fix padding value for PAD op and tensor writing to npy for FP8

Signed-off-by: Won Jeon <won.jeon@arm.com>
Change-Id: I55f663c19a1d2579d24b25c7f0d476e56e7e6dd2

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 3bff047..b6ad704 100644
--- a/reference_model/src/ops/data_layout.cc
+++ b/reference_model/src/ops/data_layout.cc
@@ -183,10 +183,12 @@
         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();
             break;
         default:
-            printNodeValidationError("Unsupported data type");
+            ASSERT_MSG(false, "TOSA_REF_TYPE %s is not supported.", EnumNameTOSAREFTYPE(Dtype));
             break;
     }