diff --git a/mlir/include/mlir-c/BuiltinTypes.h b/mlir/include/mlir-c/BuiltinTypes.h index 2212087b9898f4..d698bf4764568f 100644 --- a/mlir/include/mlir-c/BuiltinTypes.h +++ b/mlir/include/mlir-c/BuiltinTypes.h @@ -139,6 +139,16 @@ MLIR_CAPI_EXPORTED bool mlirTypeIsAFloat8E4M3B11FNUZ(MlirType type); /// context. MLIR_CAPI_EXPORTED MlirType mlirFloat8E4M3B11FNUZTypeGet(MlirContext ctx); +/// Returns the typeID of an Float8E3M4 type. +MLIR_CAPI_EXPORTED MlirTypeID mlirFloat8E3M4TypeGetTypeID(void); + +/// Checks whether the given type is an f8E3M4 type. +MLIR_CAPI_EXPORTED bool mlirTypeIsAFloat8E3M4(MlirType type); + +/// Creates an f8E3M4 type in the given context. The type is owned by the +/// context. +MLIR_CAPI_EXPORTED MlirType mlirFloat8E3M4TypeGet(MlirContext ctx); + /// Returns the typeID of an BFloat16 type. MLIR_CAPI_EXPORTED MlirTypeID mlirBFloat16TypeGetTypeID(void); diff --git a/mlir/include/mlir/IR/Builders.h b/mlir/include/mlir/IR/Builders.h index 1c4d329fbf0d8d..b5962f3783924f 100644 --- a/mlir/include/mlir/IR/Builders.h +++ b/mlir/include/mlir/IR/Builders.h @@ -66,6 +66,7 @@ class Builder { FloatType getFloat8E5M2FNUZType(); FloatType getFloat8E4M3FNUZType(); FloatType getFloat8E4M3B11FNUZType(); + FloatType getFloat8E3M4Type(); FloatType getBF16Type(); FloatType getF16Type(); FloatType getTF32Type(); diff --git a/mlir/include/mlir/IR/BuiltinTypes.h b/mlir/include/mlir/IR/BuiltinTypes.h index 4250be90ba7fb0..d12522ba55c96e 100644 --- a/mlir/include/mlir/IR/BuiltinTypes.h +++ b/mlir/include/mlir/IR/BuiltinTypes.h @@ -66,6 +66,7 @@ class FloatType : public Type { static FloatType getFloat8E5M2FNUZ(MLIRContext *ctx); static FloatType getFloat8E4M3FNUZ(MLIRContext *ctx); static FloatType getFloat8E4M3B11FNUZ(MLIRContext *ctx); + static FloatType getFloat8E3M4(MLIRContext *ctx); /// Methods for support type inquiry through isa, cast, and dyn_cast. static bool classof(Type type); @@ -411,10 +412,11 @@ inline bool BaseMemRefType::isValidElementType(Type type) { } inline bool FloatType::classof(Type type) { - return llvm::isa< - Float8E5M2Type, Float8E4M3Type, Float8E4M3FNType, Float8E5M2FNUZType, - Float8E4M3FNUZType, Float8E4M3B11FNUZType, BFloat16Type, Float16Type, - FloatTF32Type, Float32Type, Float64Type, Float80Type, Float128Type>(type); + return llvm::isa(type); } inline FloatType FloatType::getFloat8E5M2(MLIRContext *ctx) { @@ -441,6 +443,10 @@ inline FloatType FloatType::getFloat8E4M3B11FNUZ(MLIRContext *ctx) { return Float8E4M3B11FNUZType::get(ctx); } +inline FloatType FloatType::getFloat8E3M4(MLIRContext *ctx) { + return Float8E3M4Type::get(ctx); +} + inline FloatType FloatType::getBF16(MLIRContext *ctx) { return BFloat16Type::get(ctx); } diff --git a/mlir/include/mlir/IR/BuiltinTypes.td b/mlir/include/mlir/IR/BuiltinTypes.td index 176a167a3ca31c..365edcf68d8b94 100644 --- a/mlir/include/mlir/IR/BuiltinTypes.td +++ b/mlir/include/mlir/IR/BuiltinTypes.td @@ -213,6 +213,25 @@ def Builtin_Float8E4M3B11FNUZ : Builtin_FloatType<"Float8E4M3B11FNUZ", "f8E4M3B1 }]; } +//===----------------------------------------------------------------------===// +// Float8E3M4Type + +def Builtin_Float8E3M4 : Builtin_FloatType<"Float8E3M4", "f8E3M4"> { + let summary = "8-bit floating point with 3 bits exponent and 4 bit mantissa"; + let description = [{ + An 8-bit floating point type with 1 sign bit, 3 bits exponent and 4 bits + mantissa. This is not a standard type as defined by IEEE-754, but it + follows similar conventions with the following characteristics: + + * bit encoding: S1E3M4 + * exponent bias: 3 + * infinities: supported with exponent set to all 1s and mantissa 0s + * NaNs: supported with exponent bits set to all 1s and mantissa values of + {0,1}⁴ except S.111.0000 + * denormals when exponent is 0 + }]; +} + //===----------------------------------------------------------------------===// // BFloat16Type diff --git a/mlir/include/mlir/IR/CommonTypeConstraints.td b/mlir/include/mlir/IR/CommonTypeConstraints.td index c23d2d87e080f4..5b6ec167fa2420 100644 --- a/mlir/include/mlir/IR/CommonTypeConstraints.td +++ b/mlir/include/mlir/IR/CommonTypeConstraints.td @@ -342,6 +342,8 @@ def F8E4M3B11FNUZ : Type, "f8E4M3B11FNUZ t BuildableType<"$_builder.getFloat8E4M3B11FNUZType()">; def F8E5M2FNUZ : Type, "f8E5M2FNUZ type">, BuildableType<"$_builder.getFloat8E5M2FNUZType()">; +def F8E3M4 : Type, "f8E3M4 type">, + BuildableType<"$_builder.getFloat8E3M4Type()">; def AnyComplex : Type($_self)">, "complex-type", "::mlir::ComplexType">; diff --git a/mlir/include/mlir/IR/Types.h b/mlir/include/mlir/IR/Types.h index a32de33114e405..60dc8fee0f4a96 100644 --- a/mlir/include/mlir/IR/Types.h +++ b/mlir/include/mlir/IR/Types.h @@ -131,6 +131,7 @@ class Type { bool isFloat8E5M2FNUZ() const; bool isFloat8E4M3FNUZ() const; bool isFloat8E4M3B11FNUZ() const; + bool isFloat8E3M4() const; bool isBF16() const; bool isF16() const; bool isTF32() const; diff --git a/mlir/lib/AsmParser/TokenKinds.def b/mlir/lib/AsmParser/TokenKinds.def index eb3154c6da42eb..4c1c1c21031c88 100644 --- a/mlir/lib/AsmParser/TokenKinds.def +++ b/mlir/lib/AsmParser/TokenKinds.def @@ -100,6 +100,7 @@ TOK_KEYWORD(f8E4M3FN) TOK_KEYWORD(f8E5M2FNUZ) TOK_KEYWORD(f8E4M3FNUZ) TOK_KEYWORD(f8E4M3B11FNUZ) +TOK_KEYWORD(f8E3M4) TOK_KEYWORD(f128) TOK_KEYWORD(false) TOK_KEYWORD(floordiv) diff --git a/mlir/lib/AsmParser/TypeParser.cpp b/mlir/lib/AsmParser/TypeParser.cpp index 467b5f0844ab32..542eaeefe57f12 100644 --- a/mlir/lib/AsmParser/TypeParser.cpp +++ b/mlir/lib/AsmParser/TypeParser.cpp @@ -45,6 +45,7 @@ OptionalParseResult Parser::parseOptionalType(Type &type) { case Token::kw_f8E5M2FNUZ: case Token::kw_f8E4M3FNUZ: case Token::kw_f8E4M3B11FNUZ: + case Token::kw_f8E3M4: case Token::kw_bf16: case Token::kw_f16: case Token::kw_tf32: @@ -320,6 +321,9 @@ Type Parser::parseNonFunctionType() { case Token::kw_f8E4M3B11FNUZ: consumeToken(Token::kw_f8E4M3B11FNUZ); return builder.getFloat8E4M3B11FNUZType(); + case Token::kw_f8E3M4: + consumeToken(Token::kw_f8E3M4); + return builder.getFloat8E3M4Type(); case Token::kw_bf16: consumeToken(Token::kw_bf16); return builder.getBF16Type(); diff --git a/mlir/lib/Bindings/Python/IRTypes.cpp b/mlir/lib/Bindings/Python/IRTypes.cpp index 5e0aebc03e2c14..c3d42c0ef8e3cb 100644 --- a/mlir/lib/Bindings/Python/IRTypes.cpp +++ b/mlir/lib/Bindings/Python/IRTypes.cpp @@ -246,6 +246,26 @@ class PyFloat8E5M2FNUZType } }; +/// Floating Point Type subclass - Float8E3M4Type. +class PyFloat8E3M4Type : public PyConcreteType { +public: + static constexpr IsAFunctionTy isaFunction = mlirTypeIsAFloat8E3M4; + static constexpr GetTypeIDFunctionTy getTypeIdFunction = + mlirFloat8E3M4TypeGetTypeID; + static constexpr const char *pyClassName = "Float8E3M4Type"; + using PyConcreteType::PyConcreteType; + + static void bindDerived(ClassTy &c) { + c.def_static( + "get", + [](DefaultingPyMlirContext context) { + MlirType t = mlirFloat8E3M4TypeGet(context->get()); + return PyFloat8E3M4Type(context->getRef(), t); + }, + py::arg("context") = py::none(), "Create a float8_e3m4 type."); + } +}; + /// Floating Point Type subclass - BF16Type. class PyBF16Type : public PyConcreteType { public: @@ -864,6 +884,7 @@ void mlir::python::populateIRTypes(py::module &m) { PyFloat8E4M3FNUZType::bind(m); PyFloat8E4M3B11FNUZType::bind(m); PyFloat8E5M2FNUZType::bind(m); + PyFloat8E3M4Type::bind(m); PyBF16Type::bind(m); PyF16Type::bind(m); PyTF32Type::bind(m); diff --git a/mlir/lib/CAPI/IR/BuiltinTypes.cpp b/mlir/lib/CAPI/IR/BuiltinTypes.cpp index d507027357c262..2aa2e922f2abcc 100644 --- a/mlir/lib/CAPI/IR/BuiltinTypes.cpp +++ b/mlir/lib/CAPI/IR/BuiltinTypes.cpp @@ -157,6 +157,18 @@ MlirType mlirFloat8E4M3B11FNUZTypeGet(MlirContext ctx) { return wrap(FloatType::getFloat8E4M3B11FNUZ(unwrap(ctx))); } +MlirTypeID mlirFloat8E3M4TypeGetTypeID() { + return wrap(Float8E3M4Type::getTypeID()); +} + +bool mlirTypeIsAFloat8E3M4(MlirType type) { + return unwrap(type).isFloat8E3M4(); +} + +MlirType mlirFloat8E3M4TypeGet(MlirContext ctx) { + return wrap(FloatType::getFloat8E3M4(unwrap(ctx))); +} + MlirTypeID mlirBFloat16TypeGetTypeID() { return wrap(BFloat16Type::getTypeID()); } diff --git a/mlir/lib/Conversion/LLVMCommon/TypeConverter.cpp b/mlir/lib/Conversion/LLVMCommon/TypeConverter.cpp index d5df960928afb7..784deaac5ee65b 100644 --- a/mlir/lib/Conversion/LLVMCommon/TypeConverter.cpp +++ b/mlir/lib/Conversion/LLVMCommon/TypeConverter.cpp @@ -249,7 +249,7 @@ Type LLVMTypeConverter::convertIntegerType(IntegerType type) const { Type LLVMTypeConverter::convertFloatType(FloatType type) const { if (type.isFloat8E5M2() || type.isFloat8E4M3() || type.isFloat8E4M3FN() || type.isFloat8E5M2FNUZ() || type.isFloat8E4M3FNUZ() || - type.isFloat8E4M3B11FNUZ()) + type.isFloat8E4M3B11FNUZ() || type.isFloat8E3M4()) return IntegerType::get(&getContext(), type.getWidth()); return type; } diff --git a/mlir/lib/Dialect/Arith/Transforms/EmulateUnsupportedFloats.cpp b/mlir/lib/Dialect/Arith/Transforms/EmulateUnsupportedFloats.cpp index a362c8500aa5b8..51f229ef937c40 100644 --- a/mlir/lib/Dialect/Arith/Transforms/EmulateUnsupportedFloats.cpp +++ b/mlir/lib/Dialect/Arith/Transforms/EmulateUnsupportedFloats.cpp @@ -60,6 +60,7 @@ static std::optional parseFloatType(MLIRContext *ctx, .Case("f8E4M3FN", b.getFloat8E4M3FNType()) .Case("f8E5M2FNUZ", b.getFloat8E5M2FNUZType()) .Case("f8E4M3FNUZ", b.getFloat8E4M3FNUZType()) + .Case("f8E3M4", b.getFloat8E3M4Type()) .Case("bf16", b.getBF16Type()) .Case("f16", b.getF16Type()) .Case("f32", b.getF32Type()) diff --git a/mlir/lib/IR/AsmPrinter.cpp b/mlir/lib/IR/AsmPrinter.cpp index e5b1291afce2bc..02acc8c3f4659e 100644 --- a/mlir/lib/IR/AsmPrinter.cpp +++ b/mlir/lib/IR/AsmPrinter.cpp @@ -2581,6 +2581,7 @@ void AsmPrinter::Impl::printTypeImpl(Type type) { .Case([&](Type) { os << "f8E5M2FNUZ"; }) .Case([&](Type) { os << "f8E4M3FNUZ"; }) .Case([&](Type) { os << "f8E4M3B11FNUZ"; }) + .Case([&](Type) { os << "f8E3M4"; }) .Case([&](Type) { os << "bf16"; }) .Case([&](Type) { os << "f16"; }) .Case([&](Type) { os << "tf32"; }) diff --git a/mlir/lib/IR/Builders.cpp b/mlir/lib/IR/Builders.cpp index d0eb2d8fbae9d2..e3d6d71fb61dfb 100644 --- a/mlir/lib/IR/Builders.cpp +++ b/mlir/lib/IR/Builders.cpp @@ -58,6 +58,10 @@ FloatType Builder::getFloat8E4M3B11FNUZType() { return FloatType::getFloat8E4M3B11FNUZ(context); } +FloatType Builder::getFloat8E3M4Type() { + return FloatType::getFloat8E3M4(context); +} + FloatType Builder::getBF16Type() { return FloatType::getBF16(context); } FloatType Builder::getF16Type() { return FloatType::getF16(context); } diff --git a/mlir/lib/IR/BuiltinTypes.cpp b/mlir/lib/IR/BuiltinTypes.cpp index faa944937e007a..a3f5ece8c17369 100644 --- a/mlir/lib/IR/BuiltinTypes.cpp +++ b/mlir/lib/IR/BuiltinTypes.cpp @@ -88,8 +88,8 @@ IntegerType IntegerType::scaleElementBitwidth(unsigned scale) { unsigned FloatType::getWidth() { if (llvm::isa( - *this)) + Float8E5M2FNUZType, Float8E4M3FNUZType, Float8E4M3B11FNUZType, + Float8E3M4Type>(*this)) return 8; if (llvm::isa(*this)) return 16; @@ -118,6 +118,8 @@ const llvm::fltSemantics &FloatType::getFloatSemantics() { return APFloat::Float8E4M3FNUZ(); if (llvm::isa(*this)) return APFloat::Float8E4M3B11FNUZ(); + if (llvm::isa(*this)) + return APFloat::Float8E3M4(); if (llvm::isa(*this)) return APFloat::BFloat(); if (llvm::isa(*this)) diff --git a/mlir/lib/IR/MLIRContext.cpp b/mlir/lib/IR/MLIRContext.cpp index 12336701c9ca06..5c93747438ecdb 100644 --- a/mlir/lib/IR/MLIRContext.cpp +++ b/mlir/lib/IR/MLIRContext.cpp @@ -227,6 +227,7 @@ class MLIRContextImpl { Float8E5M2FNUZType f8E5M2FNUZTy; Float8E4M3FNUZType f8E4M3FNUZTy; Float8E4M3B11FNUZType f8E4M3B11FNUZTy; + Float8E3M4Type f8E3M4Ty; BFloat16Type bf16Ty; Float16Type f16Ty; FloatTF32Type tf32Ty; @@ -318,6 +319,7 @@ MLIRContext::MLIRContext(const DialectRegistry ®istry, Threading setting) impl->f8E5M2FNUZTy = TypeUniquer::get(this); impl->f8E4M3FNUZTy = TypeUniquer::get(this); impl->f8E4M3B11FNUZTy = TypeUniquer::get(this); + impl->f8E3M4Ty = TypeUniquer::get(this); impl->bf16Ty = TypeUniquer::get(this); impl->f16Ty = TypeUniquer::get(this); impl->tf32Ty = TypeUniquer::get(this); @@ -1029,6 +1031,9 @@ Float8E4M3FNUZType Float8E4M3FNUZType::get(MLIRContext *context) { Float8E4M3B11FNUZType Float8E4M3B11FNUZType::get(MLIRContext *context) { return context->getImpl().f8E4M3B11FNUZTy; } +Float8E3M4Type Float8E3M4Type::get(MLIRContext *context) { + return context->getImpl().f8E3M4Ty; +} BFloat16Type BFloat16Type::get(MLIRContext *context) { return context->getImpl().bf16Ty; } diff --git a/mlir/lib/IR/Types.cpp b/mlir/lib/IR/Types.cpp index e8cd28bf9e85de..2bc26388b6218a 100644 --- a/mlir/lib/IR/Types.cpp +++ b/mlir/lib/IR/Types.cpp @@ -46,6 +46,7 @@ bool Type::isFloat8E4M3FNUZ() const { bool Type::isFloat8E4M3B11FNUZ() const { return llvm::isa(*this); } +bool Type::isFloat8E3M4() const { return llvm::isa(*this); } bool Type::isBF16() const { return llvm::isa(*this); } bool Type::isF16() const { return llvm::isa(*this); } bool Type::isTF32() const { return llvm::isa(*this); } diff --git a/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi b/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi index 224e77a3f46be1..e3599d3c84ffed 100644 --- a/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi +++ b/mlir/python/mlir/_mlir_libs/_mlir/ir.pyi @@ -120,6 +120,7 @@ __all__ = [ "F32Type", "F64Type", "FlatSymbolRefAttr", + "Float8E3M4Type", "Float8E4M3B11FNUZType", "Float8E4M3FNType", "Float8E4M3FNUZType", @@ -1537,6 +1538,19 @@ class FlatSymbolRefAttr(Attribute): Returns the value of the FlatSymbolRef attribute as a string """ +class Float8E3M4Type(FloatType): + static_typeid: ClassVar[TypeID] + @staticmethod + def get(context: Optional[Context] = None) -> Float8E3M4Type: + """ + Create a float8_e3m4 type. + """ + @staticmethod + def isinstance(other: Type) -> bool: ... + def __init__(self, cast_from_type: Type) -> None: ... + @property + def typeid(self) -> TypeID: ... + class Float8E4M3B11FNUZType(FloatType): static_typeid: ClassVar[TypeID] @staticmethod diff --git a/mlir/python/mlir/extras/types.py b/mlir/python/mlir/extras/types.py index fde9909a8f9d6c..fe7c3e25d16906 100644 --- a/mlir/python/mlir/extras/types.py +++ b/mlir/python/mlir/extras/types.py @@ -12,6 +12,7 @@ F16Type, F32Type, F64Type, + Float8E3M4Type, Float8E4M3B11FNUZType, Float8E4M3FNType, Float8E4M3Type, @@ -72,6 +73,7 @@ def ui(width): f8E4M3 = lambda: Float8E4M3Type.get() f8E4M3FN = lambda: Float8E4M3FNType.get() f8E4M3B11FNUZ = lambda: Float8E4M3B11FNUZType.get() +f8E3M4 = lambda: Float8E3M4Type.get() none = lambda: NoneType.get() diff --git a/mlir/test/IR/attribute.mlir b/mlir/test/IR/attribute.mlir index 362e98134ee4a6..ac0aec113add17 100644 --- a/mlir/test/IR/attribute.mlir +++ b/mlir/test/IR/attribute.mlir @@ -60,6 +60,10 @@ func.func @float_attrs_pass() { // CHECK: float_attr = 2.000000e+00 : f8E4M3B11FNUZ float_attr = 2. : f8E4M3B11FNUZ } : () -> () + "test.float_attrs"() { + // CHECK: float_attr = 2.000000e+00 : f8E3M4 + float_attr = 2. : f8E3M4 + } : () -> () "test.float_attrs"() { // CHECK: float_attr = 2.000000e+00 : f16 float_attr = 2. : f16 diff --git a/mlir/test/Target/LLVMIR/llvmir.mlir b/mlir/test/Target/LLVMIR/llvmir.mlir index db54d131299c66..82256f753abdd0 100644 --- a/mlir/test/Target/LLVMIR/llvmir.mlir +++ b/mlir/test/Target/LLVMIR/llvmir.mlir @@ -39,6 +39,9 @@ llvm.mlir.global internal constant @string_const("foobar") : !llvm.array<6 x i8> // CHECK: @int_global_undef = internal global i64 undef llvm.mlir.global internal @int_global_undef() : i64 +// CHECK: @f8E3M4_global_as_i8 = internal global i8 56 +llvm.mlir.global internal @f8E3M4_global_as_i8(1.5 : f8E3M4) : i8 + // CHECK: @f8E4M3_global_as_i8 = internal global i8 60 llvm.mlir.global internal @f8E4M3_global_as_i8(1.5 : f8E4M3) : i8 diff --git a/mlir/test/python/ir/builtin_types.py b/mlir/test/python/ir/builtin_types.py index 3178f58cf2e74d..2161f110ac31e2 100644 --- a/mlir/test/python/ir/builtin_types.py +++ b/mlir/test/python/ir/builtin_types.py @@ -113,6 +113,8 @@ def testTypeIsInstance(): def testFloatTypeSubclasses(): ctx = Context() # CHECK: True + print(isinstance(Type.parse("f8E3M4", ctx), FloatType)) + # CHECK: True print(isinstance(Type.parse("f8E4M3", ctx), FloatType)) # CHECK: True print(isinstance(Type.parse("f8E4M3FN", ctx), FloatType)) @@ -231,6 +233,8 @@ def testIndexType(): @run def testFloatType(): with Context(): + # CHECK: float: f8E3M4 + print("float:", Float8E3M4Type.get()) # CHECK: float: f8E4M3 print("float:", Float8E4M3Type.get()) # CHECK: float: f8E4M3FN @@ -605,6 +609,7 @@ def testTypeIDs(): types = [ (IntegerType, IntegerType.get_signless(16)), (IndexType, IndexType.get()), + (Float8E3M4Type, Float8E3M4Type.get()), (Float8E4M3Type, Float8E4M3Type.get()), (Float8E4M3FNType, Float8E4M3FNType.get()), (Float8E5M2Type, Float8E5M2Type.get()), @@ -629,6 +634,7 @@ def testTypeIDs(): # CHECK: IntegerType(i16) # CHECK: IndexType(index) + # CHECK: Float8E3M4Type(f8E3M4) # CHECK: Float8E4M3Type(f8E4M3) # CHECK: Float8E4M3FNType(f8E4M3FN) # CHECK: Float8E5M2Type(f8E5M2) @@ -707,6 +713,9 @@ def print_downcasted(typ): # CHECK: F64Type # CHECK: F64Type(f64) print_downcasted(F64Type.get()) + # CHECK: Float8E3M4Type + # CHECK: Float8E3M4Type(f8E3M4) + print_downcasted(Float8E3M4Type.get()) # CHECK: Float8E4M3B11FNUZType # CHECK: Float8E4M3B11FNUZType(f8E4M3B11FNUZ) print_downcasted(Float8E4M3B11FNUZType.get()) diff --git a/mlir/utils/lldb-scripts/mlirDataFormatters.py b/mlir/utils/lldb-scripts/mlirDataFormatters.py index ed0ee431fd7d81..e7c526842439b9 100644 --- a/mlir/utils/lldb-scripts/mlirDataFormatters.py +++ b/mlir/utils/lldb-scripts/mlirDataFormatters.py @@ -56,6 +56,7 @@ def build_ptr_str_from_addr(addrValue: lldb.SBValue, type: lldb.SBType): "mlir::Float8E5M2FNUZType": '"f8E5M2FNUZ"', "mlir::Float8E4M3FNUZType": '"f8E4M3FNUZ"', "mlir::Float8E4M3B11FNUZType": '"f8E4M3B11FNUZ"', + "mlir::Float8E3M4Type": '"f8E3M4"', "mlir::BFloat16Type": '"bf16"', "mlir::Float16Type": '"f16"', "mlir::FloatTF32Type": '"tf32"', diff --git a/mlir/utils/tree-sitter-mlir/grammar.js b/mlir/utils/tree-sitter-mlir/grammar.js index a657874f894b7e..b5926d75da4f25 100644 --- a/mlir/utils/tree-sitter-mlir/grammar.js +++ b/mlir/utils/tree-sitter-mlir/grammar.js @@ -230,8 +230,8 @@ const common = { integer_type : $ => token(seq(choice('si', 'ui', 'i'), /[1-9]/, repeat(/[0-9]/))), float_type : $ => token( - choice('f16', 'f32', 'f64', 'f80', 'f128', 'bf16', 'f8E4M3FN', 'f8E4M3', - 'f8E5M2')), + choice('f16', 'f32', 'f64', 'f80', 'f128', 'bf16', 'f8E3M4', 'f8E4M3FN', + 'f8E4M3', 'f8E5M2')), index_type : $ => token('index'), none_type : $ => token('none'), complex_type : $ => seq(token('complex'), '<', $._prim_type, '>'),