Add bfloat16 to scalar_type.fbs (pytorch#16882)

lucylq · Github Executorch · web-flow · commit 61ae2f6261b6 · 2026-01-26T13:34:55.000-08:00
### Summary This should cover all the scalar_type.fbs files: https://github.com/search?q=repo%3Apytorch%2Fexecutorch%20path%3Ascalar_type.fbs&type=code ### Test plan Deserialize a program with bfloat16. See: https://www.internalfb.com/manifold/explorer/executorch/tree/models/whisper/test/model.pte ``` >>> from executorch.exir._serialize import _deserialize_pte_binary >>> f = open("model.pte", "rb") >>> m = _deserialize_pte_binary(f.read()) >>> ``` Co-authored-by: Github Executorch <github_executorch@arm.com>
diff --git a/devtools/bundled_program/schema/scalar_type.fbs b/devtools/bundled_program/schema/scalar_type.fbs
@@ -21,6 +21,7 @@ enum ScalarType : byte {
   QINT8 = 12,
   QUINT8 = 13,
   QINT32 = 14,
+  BFLOAT16 = 15,
   QUINT4X2 = 16,
   QUINT2X4 = 17,
   BITS16 = 22,
@@ -35,7 +36,6 @@ enum ScalarType : byte {
   // COMPLEXHALF = 8,
   // COMPLEXFLOAT = 9,
   // COMPLEXDOUBLE = 10,
-  // BFLOAT16 = 15,
   // BITS1x8 = 18,
   // BITS2x4 = 19,
   // BITS4x2 = 20,
diff --git a/devtools/etdump/scalar_type.fbs b/devtools/etdump/scalar_type.fbs
@@ -21,6 +21,7 @@ enum ScalarType : byte {
   QINT8 = 12,
   QUINT8 = 13,
   QINT32 = 14,
+  BFLOAT16 = 15,
   QUINT4X2 = 16,
   QUINT2X4 = 17,
   BITS16 = 22,
@@ -35,7 +36,6 @@ enum ScalarType : byte {
   // COMPLEXHALF = 8,
   // COMPLEXFLOAT = 9,
   // COMPLEXDOUBLE = 10,
-  // BFLOAT16 = 15,
   // BITS1x8 = 18,
   // BITS2x4 = 19,
   // BITS4x2 = 20,
diff --git a/extension/flat_tensor/serialize/scalar_type.fbs b/extension/flat_tensor/serialize/scalar_type.fbs
@@ -23,14 +23,14 @@ enum ScalarType : byte {
   QINT8 = 12,
   QUINT8 = 13,
   QINT32 = 14,
+  BFLOAT16 = 15,
   QUINT4X2 = 16,
   QUINT2X4 = 17,
   BITS16 = 22,
   // Types currently not implemented.
   // COMPLEXHALF = 8,
   // COMPLEXFLOAT = 9,
   // COMPLEXDOUBLE = 10,
-  // BFLOAT16 = 15,
   // BITS1x8 = 18,
   // BITS2x4 = 19,
   // BITS4x2 = 20,
diff --git a/schema/scalar_type.fbs b/schema/scalar_type.fbs
@@ -21,6 +21,7 @@ enum ScalarType : byte {
   QINT8 = 12,
   QUINT8 = 13,
   QINT32 = 14,
+  BFLOAT16 = 15,
   QUINT4X2 = 16,
   QUINT2X4 = 17,
   BITS16 = 22,
@@ -35,7 +36,6 @@ enum ScalarType : byte {
   // COMPLEXHALF = 8,
   // COMPLEXFLOAT = 9,
   // COMPLEXDOUBLE = 10,
-  // BFLOAT16 = 15,
   // BITS1x8 = 18,
   // BITS2x4 = 19,
   // BITS4x2 = 20,
