commit c53436e4933c50c7465712590778ca283e8c07c5
author Eric Kunze <eric.kunze@arm.com> Tue Mar 19 13:55:38 2024 -0700
committer Eric Kunze <eric.kunze@arm.com> Tue Mar 19 21:03:04 2024 +0000
tree c81040bbd5d7203082ef3b7e038b3058a619b2f0
parent 14e33564c7457ef0c0d0f5e6577c1036c0471866

Revert "[serialization_lib] Add support for FP8E4M3 and FP8E5M2"

This reverts commit a029f1f02707f40f6990df53fd4f56684490d58f.

This commit was not meant to be included in the 0.90 branch.

Change-Id: I5b252f2603a7554efab2b997faaf2b59d62ee3cd

include/tosa_generated.h

diff --git a/include/tosa_generated.h b/include/tosa_generated.h
index b5a8bd5..2ecd35a 100644
--- a/include/tosa_generated.h
+++ b/include/tosa_generated.h
@@ -116,13 +116,11 @@
   DType_FP16 = 10,
   DType_BF16 = 11,
   DType_SHAPE = 12,
-  DType_FP8E4M3 = 13,
-  DType_FP8E5M2 = 14,
   DType_MIN = DType_UNKNOWN,
-  DType_MAX = DType_FP8E5M2
+  DType_MAX = DType_SHAPE
 };
 
-inline const DType (&EnumValuesDType())[15] {
+inline const DType (&EnumValuesDType())[13] {
   static const DType values[] = {
     DType_UNKNOWN,
     DType_BOOL,
@@ -136,15 +134,13 @@
     DType_UINT16,
     DType_FP16,
     DType_BF16,
-    DType_SHAPE,
-    DType_FP8E4M3,
-    DType_FP8E5M2
+    DType_SHAPE
   };
   return values;
 }
 
 inline const char * const *EnumNamesDType() {
-  static const char * const names[16] = {
+  static const char * const names[14] = {
     "UNKNOWN",
     "BOOL",
     "UINT8",
@@ -158,15 +154,13 @@
     "FP16",
     "BF16",
     "SHAPE",
-    "FP8E4M3",
-    "FP8E5M2",
     nullptr
   };
   return names;
 }
 
 inline const char *EnumNameDType(DType e) {
-  if (::flatbuffers::IsOutRange(e, DType_UNKNOWN, DType_FP8E5M2)) return "";
+  if (::flatbuffers::IsOutRange(e, DType_UNKNOWN, DType_SHAPE)) return "";
   const size_t index = static_cast<size_t>(e);
   return EnumNamesDType()[index];
 }

python/serializer/tosa_serializer.py

diff --git a/python/serializer/tosa_serializer.py b/python/serializer/tosa_serializer.py
index 1aadbff..c44b225 100644
--- a/python/serializer/tosa_serializer.py
+++ b/python/serializer/tosa_serializer.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2020-2024, ARM Limited.
+# Copyright (c) 2020-2023, ARM Limited.
 #
 #    Licensed under the Apache License, Version 2.0 (the "License");
 #    you may not use this file except in compliance with the License.
@@ -63,8 +63,6 @@
     "FP16",
     "BF16",
     "SHAPE",
-    "FP8E4M3",
-    "FP8E5M2",
 ]
 
 ByteMask = np.uint64(0xFF)
@@ -427,12 +425,7 @@
         self.shape = shape
         self.dtype = dtype
 
-        if (
-            dtype == DType.FP32
-            or dtype == DType.BF16
-            or dtype == DType.FP8E4M3
-            or dtype == DType.FP8E5M2
-        ):
+        if dtype == DType.FP32 or dtype == DType.BF16:
             fntype = np.float32
         elif dtype == DType.FP16:
             fntype = np.float16
@@ -532,12 +525,7 @@
             elif self.dtype == DType.FP16:
                 np_arr = np.array(self.data, dtype=np.float16)
                 u8_data.extend(np_arr.view(np.uint8))
-            elif (
-                self.dtype == DType.FP32
-                or self.dtype == DType.BF16
-                or self.dtype == DType.FP8E4M3
-                or self.dtype == DType.FP8E5M2
-            ):
+            elif self.dtype == DType.FP32 or self.dtype == DType.BF16:
                 # for val in self.data:
                 #     b = struct.pack("!f", val)
                 #     u8_data.extend([b[3], b[2], b[1], b[0]])

python/tosa/DType.py

diff --git a/python/tosa/DType.py b/python/tosa/DType.py
index e585cb9..6df2dcb 100644
--- a/python/tosa/DType.py
+++ b/python/tosa/DType.py
@@ -16,5 +16,3 @@
     FP16 = 10
     BF16 = 11
     SHAPE = 12
-    FP8E4M3 = 13
-    FP8E5M2 = 14

schema/tosa.fbs

diff --git a/schema/tosa.fbs b/schema/tosa.fbs
index 1c2a85e..171818d 100644
--- a/schema/tosa.fbs
+++ b/schema/tosa.fbs
@@ -37,8 +37,6 @@
   FP16,
   BF16,
   SHAPE,
-  FP8E4M3,
-  FP8E5M2,
 }
 
 enum ResizeMode:uint32 {