[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