[Mlir-commits] [mlir] [mlir][tosa] Add FP8 support (PR #127730)

Luke Hutton llvmlistbot at llvm.org
Wed Feb 19 08:47:00 PST 2025


================
@@ -74,16 +74,25 @@ def Tosa_QuantizedInt : AnyTypeOf<[Tosa_QuantizedType<"uint8", [8], 0>,
                                    Tosa_QuantizedType<"int16", [16, 0], 1>,
                                    Tosa_QuantizedType<"int32", [32, 0], 1>]>;
 
+def Tosa_F8 : AnyTypeOf<[
+                        F8E4M3FN,
+                        F8E5M2]>;
+
 //===----------------------------------------------------------------------===//
 // Multi-category types.
 //===----------------------------------------------------------------------===//
 def Tosa_AnyNumber : AnyTypeOf<[Tosa_Int, Tosa_QuantizedInt, AnyFloat],
                                 "number">;
 
+// Add F8 type support to Tosa_AnyNumber
+def Tosa_AnyNumber_Extended : AnyTypeOf<[Tosa_AnyNumber, Tosa_F8],
----------------
lhutton1 wrote:

`Tosa_AnyNumber` already uses `AnyFloat` so I believe it could already support FP8?

https://github.com/llvm/llvm-project/pull/127730


More information about the Mlir-commits mailing list