[Mlir-commits] [mlir] 1f515f3 - [mlir] Add tablegen types for fp8 types

[Benjamin Kramer]

Author: Benjamin Kramer
Date: 2022-11-22T20:13:08+01:00
New Revision: 1f515f30d0083ec0d74f0d9885e614f758f12687

URL: https://github.com/llvm/llvm-project/commit/1f515f30d0083ec0d74f0d9885e614f758f12687
DIFF: https://github.com/llvm/llvm-project/commit/1f515f30d0083ec0d74f0d9885e614f758f12687.diff

LOG: [mlir] Add tablegen types for fp8 types

For consistency and allowing users easy access to these types.

Differential Revision: https://reviews.llvm.org/D138516

Added: 
    

Modified: 
    mlir/include/mlir/IR/OpBase.td

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir/IR/OpBase.td b/mlir/include/mlir/IR/OpBase.td
index da0e2de6bd2f..9c87071ee8eb 100644
--- a/mlir/include/mlir/IR/OpBase.td
+++ b/mlir/include/mlir/IR/OpBase.td
@@ -490,6 +490,10 @@ def F128 : F<128>;
 
 def BF16 : Type<CPred<"$_self.isBF16()">, "bfloat16 type">,
            BuildableType<"$_builder.getBF16Type()">;
+def F8E4M3FN : Type<CPred<"$_self.isFloat8E4M3FN()">, "f8E4M3FN type">,
+               BuildableType<"$_builder.getFloat8E4M3FNType()">;
+def F8E5M2 : Type<CPred<"$_self.isFloat8E5M2()">, "f8E5M2 type">,
+             BuildableType<"$_builder.getFloat8E5M2Type()">;
 
 def AnyComplex : Type<CPred<"$_self.isa<::mlir::ComplexType>()">,
                       "complex-type", "::mlir::ComplexType">;