Skip to content

Commit 73e9e3a

Browse files
apivovarovyuxuanchen1997
apivovarov
authored and
yuxuanchen1997
committed
[MLIR] Add f8E4M3 IEEE 754 type (#97118)
Summary: This PR adds `f8E4M3` type to mlir. `f8E4M3` type follows IEEE 754 convention ```c f8E4M3 (IEEE 754) - Exponent bias: 7 - Maximum stored exponent value: 14 (binary 1110) - Maximum unbiased exponent value: 14 - 7 = 7 - Minimum stored exponent value: 1 (binary 0001) - Minimum unbiased exponent value: 1 − 7 = −6 - Precision specifies the total number of bits used for the significand (mantisa), including implicit leading integer bit = 3 + 1 = 4 - Follows IEEE 754 conventions for representation of special values - Has Positive and Negative zero - Has Positive and Negative infinity - Has NaNs Additional details: - Max exp (unbiased): 7 - Min exp (unbiased): -6 - Infinities (+/-): S.1111.000 - Zeros (+/-): S.0000.000 - NaNs: S.1111.{001, 010, 011, 100, 101, 110, 111} - Max normal number: S.1110.111 = +/-2^(7) x (1 + 0.875) = +/-240 - Min normal number: S.0001.000 = +/-2^(-6) - Max subnormal number: S.0000.111 = +/-2^(-6) x 0.875 = +/-2^(-9) x 7 - Min subnormal number: S.0000.001 = +/-2^(-6) x 0.125 = +/-2^(-9) ``` Related PRs: - [PR-97179](#97179) [APFloat] Add support for f8E4M3 IEEE 754 type Test Plan: Reviewers: Subscribers: Tasks: Tags: Differential Revision: https://phabricator.intern.facebook.com/D60251173
1 parent 8207ae2 commit 73e9e3a

File tree

24 files changed

+136
-10
lines changed

24 files changed

+136
-10
lines changed

‎mlir/include/mlir-c/BuiltinTypes.h

+10
Original file line numberDiff line numberDiff line change
@@ -89,6 +89,16 @@ MLIR_CAPI_EXPORTED bool mlirTypeIsAFloat8E5M2(MlirType type);
8989
/// context.
9090
MLIR_CAPI_EXPORTED MlirType mlirFloat8E5M2TypeGet(MlirContext ctx);
9191

92+
/// Returns the typeID of an Float8E4M3 type.
93+
MLIR_CAPI_EXPORTED MlirTypeID mlirFloat8E4M3TypeGetTypeID(void);
94+
95+
/// Checks whether the given type is an f8E4M3 type.
96+
MLIR_CAPI_EXPORTED bool mlirTypeIsAFloat8E4M3(MlirType type);
97+
98+
/// Creates an f8E4M3 type in the given context. The type is owned by the
99+
/// context.
100+
MLIR_CAPI_EXPORTED MlirType mlirFloat8E4M3TypeGet(MlirContext ctx);
101+
92102
/// Returns the typeID of an Float8E4M3FN type.
93103
MLIR_CAPI_EXPORTED MlirTypeID mlirFloat8E4M3FNTypeGetTypeID(void);
94104

‎mlir/include/mlir/IR/Builders.h

+1
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ class Builder {
6161

6262
// Types.
6363
FloatType getFloat8E5M2Type();
64+
FloatType getFloat8E4M3Type();
6465
FloatType getFloat8E4M3FNType();
6566
FloatType getFloat8E5M2FNUZType();
6667
FloatType getFloat8E4M3FNUZType();

‎mlir/include/mlir/IR/BuiltinTypes.h

+9-4
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ class FloatType : public Type {
6161
static FloatType getF80(MLIRContext *ctx);
6262
static FloatType getF128(MLIRContext *ctx);
6363
static FloatType getFloat8E5M2(MLIRContext *ctx);
64+
static FloatType getFloat8E4M3(MLIRContext *ctx);
6465
static FloatType getFloat8E4M3FN(MLIRContext *ctx);
6566
static FloatType getFloat8E5M2FNUZ(MLIRContext *ctx);
6667
static FloatType getFloat8E4M3FNUZ(MLIRContext *ctx);
@@ -410,16 +411,20 @@ inline bool BaseMemRefType::isValidElementType(Type type) {
410411
}
411412

412413
inline bool FloatType::classof(Type type) {
413-
return llvm::isa<Float8E5M2Type, Float8E4M3FNType, Float8E5M2FNUZType,
414-
Float8E4M3FNUZType, Float8E4M3B11FNUZType, BFloat16Type,
415-
Float16Type, FloatTF32Type, Float32Type, Float64Type,
416-
Float80Type, Float128Type>(type);
414+
return llvm::isa<
415+
Float8E5M2Type, Float8E4M3Type, Float8E4M3FNType, Float8E5M2FNUZType,
416+
Float8E4M3FNUZType, Float8E4M3B11FNUZType, BFloat16Type, Float16Type,
417+
FloatTF32Type, Float32Type, Float64Type, Float80Type, Float128Type>(type);
417418
}
418419

419420
inline FloatType FloatType::getFloat8E5M2(MLIRContext *ctx) {
420421
return Float8E5M2Type::get(ctx);
421422
}
422423

424+
inline FloatType FloatType::getFloat8E4M3(MLIRContext *ctx) {
425+
return Float8E4M3Type::get(ctx);
426+
}
427+
423428
inline FloatType FloatType::getFloat8E4M3FN(MLIRContext *ctx) {
424429
return Float8E4M3FNType::get(ctx);
425430
}

‎mlir/include/mlir/IR/BuiltinTypes.td

+19
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,25 @@ def Builtin_Float8E5M2 : Builtin_FloatType<"Float8E5M2", "f8E5M2"> {
106106
}];
107107
}
108108

109+
//===----------------------------------------------------------------------===//
110+
// Float8E4M3Type
111+
112+
def Builtin_Float8E4M3 : Builtin_FloatType<"Float8E4M3", "f8E4M3"> {
113+
let summary = "8-bit floating point with 3 bit mantissa";
114+
let description = [{
115+
An 8-bit floating point type with 1 sign bit, 4 bits exponent and 3 bits
116+
mantissa. This is not a standard type as defined by IEEE-754, but it
117+
follows similar conventions with the following characteristics:
118+
119+
* bit encoding: S1E4M3
120+
* exponent bias: 7
121+
* infinities: supported with exponent set to all 1s and mantissa 0s
122+
* NaNs: supported with exponent bits set to all 1s and mantissa of
123+
(001, 010, 011, 100, 101, 110, 111)
124+
* denormals when exponent is 0
125+
}];
126+
}
127+
109128
//===----------------------------------------------------------------------===//
110129
// Float8E4M3FNType
111130

‎mlir/include/mlir/IR/CommonTypeConstraints.td

+2
Original file line numberDiff line numberDiff line change
@@ -334,6 +334,8 @@ def F8E4M3FN : Type<CPred<"$_self.isFloat8E4M3FN()">, "f8E4M3FN type">,
334334
BuildableType<"$_builder.getFloat8E4M3FNType()">;
335335
def F8E5M2 : Type<CPred<"$_self.isFloat8E5M2()">, "f8E5M2 type">,
336336
BuildableType<"$_builder.getFloat8E5M2Type()">;
337+
def F8E4M3 : Type<CPred<"$_self.isFloat8E4M3()">, "f8E4M3 type">,
338+
BuildableType<"$_builder.getFloat8E4M3Type()">;
337339
def F8E4M3FNUZ : Type<CPred<"$_self.isFloat8E4M3FNUZ()">, "f8E4M3FNUZ type">,
338340
BuildableType<"$_builder.getFloat8E4M3FNUZType()">;
339341
def F8E4M3B11FNUZ : Type<CPred<"$_self.isFloat8E4M3B11FNUZ()">, "f8E4M3B11FNUZ type">,

‎mlir/include/mlir/IR/Types.h

+1
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,7 @@ class Type {
126126
// derived types should use isa/dyn_cast.
127127
bool isIndex() const;
128128
bool isFloat8E5M2() const;
129+
bool isFloat8E4M3() const;
129130
bool isFloat8E4M3FN() const;
130131
bool isFloat8E5M2FNUZ() const;
131132
bool isFloat8E4M3FNUZ() const;

‎mlir/lib/AsmParser/TokenKinds.def

+1
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,7 @@ TOK_KEYWORD(f32)
9595
TOK_KEYWORD(f64)
9696
TOK_KEYWORD(f80)
9797
TOK_KEYWORD(f8E5M2)
98+
TOK_KEYWORD(f8E4M3)
9899
TOK_KEYWORD(f8E4M3FN)
99100
TOK_KEYWORD(f8E5M2FNUZ)
100101
TOK_KEYWORD(f8E4M3FNUZ)

‎mlir/lib/AsmParser/TypeParser.cpp

+4
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ OptionalParseResult Parser::parseOptionalType(Type &type) {
4040
case Token::kw_vector:
4141
case Token::inttype:
4242
case Token::kw_f8E5M2:
43+
case Token::kw_f8E4M3:
4344
case Token::kw_f8E4M3FN:
4445
case Token::kw_f8E5M2FNUZ:
4546
case Token::kw_f8E4M3FNUZ:
@@ -304,6 +305,9 @@ Type Parser::parseNonFunctionType() {
304305
case Token::kw_f8E5M2:
305306
consumeToken(Token::kw_f8E5M2);
306307
return builder.getFloat8E5M2Type();
308+
case Token::kw_f8E4M3:
309+
consumeToken(Token::kw_f8E4M3);
310+
return builder.getFloat8E4M3Type();
307311
case Token::kw_f8E4M3FN:
308312
consumeToken(Token::kw_f8E4M3FN);
309313
return builder.getFloat8E4M3FNType();

‎mlir/lib/Bindings/Python/IRTypes.cpp

+22-1
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,7 @@ class PyFloat8E4M3FNType
143143
}
144144
};
145145

146-
/// Floating Point Type subclass - Float8M5E2Type.
146+
/// Floating Point Type subclass - Float8E5M2Type.
147147
class PyFloat8E5M2Type : public PyConcreteType<PyFloat8E5M2Type, PyFloatType> {
148148
public:
149149
static constexpr IsAFunctionTy isaFunction = mlirTypeIsAFloat8E5M2;
@@ -163,6 +163,26 @@ class PyFloat8E5M2Type : public PyConcreteType<PyFloat8E5M2Type, PyFloatType> {
163163
}
164164
};
165165

166+
/// Floating Point Type subclass - Float8E4M3Type.
167+
class PyFloat8E4M3Type : public PyConcreteType<PyFloat8E4M3Type, PyFloatType> {
168+
public:
169+
static constexpr IsAFunctionTy isaFunction = mlirTypeIsAFloat8E4M3;
170+
static constexpr GetTypeIDFunctionTy getTypeIdFunction =
171+
mlirFloat8E4M3TypeGetTypeID;
172+
static constexpr const char *pyClassName = "Float8E4M3Type";
173+
using PyConcreteType::PyConcreteType;
174+
175+
static void bindDerived(ClassTy &c) {
176+
c.def_static(
177+
"get",
178+
[](DefaultingPyMlirContext context) {
179+
MlirType t = mlirFloat8E4M3TypeGet(context->get());
180+
return PyFloat8E4M3Type(context->getRef(), t);
181+
},
182+
py::arg("context") = py::none(), "Create a float8_e4m3 type.");
183+
}
184+
};
185+
166186
/// Floating Point Type subclass - Float8E4M3FNUZ.
167187
class PyFloat8E4M3FNUZType
168188
: public PyConcreteType<PyFloat8E4M3FNUZType, PyFloatType> {
@@ -840,6 +860,7 @@ void mlir::python::populateIRTypes(py::module &m) {
840860
PyIndexType::bind(m);
841861
PyFloat8E4M3FNType::bind(m);
842862
PyFloat8E5M2Type::bind(m);
863+
PyFloat8E4M3Type::bind(m);
843864
PyFloat8E4M3FNUZType::bind(m);
844865
PyFloat8E4M3B11FNUZType::bind(m);
845866
PyFloat8E5M2FNUZType::bind(m);

‎mlir/lib/CAPI/IR/BuiltinTypes.cpp

+12
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,18 @@ MlirType mlirFloat8E5M2TypeGet(MlirContext ctx) {
9797
return wrap(FloatType::getFloat8E5M2(unwrap(ctx)));
9898
}
9999

100+
MlirTypeID mlirFloat8E4M3TypeGetTypeID() {
101+
return wrap(Float8E4M3Type::getTypeID());
102+
}
103+
104+
bool mlirTypeIsAFloat8E4M3(MlirType type) {
105+
return unwrap(type).isFloat8E4M3();
106+
}
107+
108+
MlirType mlirFloat8E4M3TypeGet(MlirContext ctx) {
109+
return wrap(FloatType::getFloat8E4M3(unwrap(ctx)));
110+
}
111+
100112
MlirTypeID mlirFloat8E4M3FNTypeGetTypeID() {
101113
return wrap(Float8E4M3FNType::getTypeID());
102114
}

‎mlir/lib/Conversion/LLVMCommon/TypeConverter.cpp

+3-2
Original file line numberDiff line numberDiff line change
@@ -247,8 +247,9 @@ Type LLVMTypeConverter::convertIntegerType(IntegerType type) const {
247247
}
248248

249249
Type LLVMTypeConverter::convertFloatType(FloatType type) const {
250-
if (type.isFloat8E5M2() || type.isFloat8E4M3FN() || type.isFloat8E5M2FNUZ() ||
251-
type.isFloat8E4M3FNUZ() || type.isFloat8E4M3B11FNUZ())
250+
if (type.isFloat8E5M2() || type.isFloat8E4M3() || type.isFloat8E4M3FN() ||
251+
type.isFloat8E5M2FNUZ() || type.isFloat8E4M3FNUZ() ||
252+
type.isFloat8E4M3B11FNUZ())
252253
return IntegerType::get(&getContext(), type.getWidth());
253254
return type;
254255
}

‎mlir/lib/Dialect/Arith/Transforms/EmulateUnsupportedFloats.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@ static std::optional<FloatType> parseFloatType(MLIRContext *ctx,
5656
Builder b(ctx);
5757
return llvm::StringSwitch<std::optional<FloatType>>(name)
5858
.Case("f8E5M2", b.getFloat8E5M2Type())
59+
.Case("f8E4M3", b.getFloat8E4M3Type())
5960
.Case("f8E4M3FN", b.getFloat8E4M3FNType())
6061
.Case("f8E5M2FNUZ", b.getFloat8E5M2FNUZType())
6162
.Case("f8E4M3FNUZ", b.getFloat8E4M3FNUZType())

‎mlir/lib/IR/AsmPrinter.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -2576,6 +2576,7 @@ void AsmPrinter::Impl::printTypeImpl(Type type) {
25762576
})
25772577
.Case<IndexType>([&](Type) { os << "index"; })
25782578
.Case<Float8E5M2Type>([&](Type) { os << "f8E5M2"; })
2579+
.Case<Float8E4M3Type>([&](Type) { os << "f8E4M3"; })
25792580
.Case<Float8E4M3FNType>([&](Type) { os << "f8E4M3FN"; })
25802581
.Case<Float8E5M2FNUZType>([&](Type) { os << "f8E5M2FNUZ"; })
25812582
.Case<Float8E4M3FNUZType>([&](Type) { os << "f8E4M3FNUZ"; })

‎mlir/lib/IR/Builders.cpp

+4
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,10 @@ FloatType Builder::getFloat8E5M2Type() {
3838
return FloatType::getFloat8E5M2(context);
3939
}
4040

41+
FloatType Builder::getFloat8E4M3Type() {
42+
return FloatType::getFloat8E4M3(context);
43+
}
44+
4145
FloatType Builder::getFloat8E4M3FNType() {
4246
return FloatType::getFloat8E4M3FN(context);
4347
}

‎mlir/lib/IR/BuiltinTypes.cpp

+5-2
Original file line numberDiff line numberDiff line change
@@ -87,8 +87,9 @@ IntegerType IntegerType::scaleElementBitwidth(unsigned scale) {
8787
//===----------------------------------------------------------------------===//
8888

8989
unsigned FloatType::getWidth() {
90-
if (llvm::isa<Float8E5M2Type, Float8E4M3FNType, Float8E5M2FNUZType,
91-
Float8E4M3FNUZType, Float8E4M3B11FNUZType>(*this))
90+
if (llvm::isa<Float8E5M2Type, Float8E4M3Type, Float8E4M3FNType,
91+
Float8E5M2FNUZType, Float8E4M3FNUZType, Float8E4M3B11FNUZType>(
92+
*this))
9293
return 8;
9394
if (llvm::isa<Float16Type, BFloat16Type>(*this))
9495
return 16;
@@ -107,6 +108,8 @@ unsigned FloatType::getWidth() {
107108
const llvm::fltSemantics &FloatType::getFloatSemantics() {
108109
if (llvm::isa<Float8E5M2Type>(*this))
109110
return APFloat::Float8E5M2();
111+
if (llvm::isa<Float8E4M3Type>(*this))
112+
return APFloat::Float8E4M3();
110113
if (llvm::isa<Float8E4M3FNType>(*this))
111114
return APFloat::Float8E4M3FN();
112115
if (llvm::isa<Float8E5M2FNUZType>(*this))

‎mlir/lib/IR/MLIRContext.cpp

+5
Original file line numberDiff line numberDiff line change
@@ -222,6 +222,7 @@ class MLIRContextImpl {
222222

223223
/// Cached Type Instances.
224224
Float8E5M2Type f8E5M2Ty;
225+
Float8E4M3Type f8E4M3Ty;
225226
Float8E4M3FNType f8E4M3FNTy;
226227
Float8E5M2FNUZType f8E5M2FNUZTy;
227228
Float8E4M3FNUZType f8E4M3FNUZTy;
@@ -312,6 +313,7 @@ MLIRContext::MLIRContext(const DialectRegistry &registry, Threading setting)
312313
//// Types.
313314
/// Floating-point Types.
314315
impl->f8E5M2Ty = TypeUniquer::get<Float8E5M2Type>(this);
316+
impl->f8E4M3Ty = TypeUniquer::get<Float8E4M3Type>(this);
315317
impl->f8E4M3FNTy = TypeUniquer::get<Float8E4M3FNType>(this);
316318
impl->f8E5M2FNUZTy = TypeUniquer::get<Float8E5M2FNUZType>(this);
317319
impl->f8E4M3FNUZTy = TypeUniquer::get<Float8E4M3FNUZType>(this);
@@ -1012,6 +1014,9 @@ StorageUniquer &MLIRContext::getTypeUniquer() { return getImpl().typeUniquer; }
10121014
Float8E5M2Type Float8E5M2Type::get(MLIRContext *context) {
10131015
return context->getImpl().f8E5M2Ty;
10141016
}
1017+
Float8E4M3Type Float8E4M3Type::get(MLIRContext *context) {
1018+
return context->getImpl().f8E4M3Ty;
1019+
}
10151020
Float8E4M3FNType Float8E4M3FNType::get(MLIRContext *context) {
10161021
return context->getImpl().f8E4M3FNTy;
10171022
}

‎mlir/lib/IR/Types.cpp

+1
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ Type AbstractType::replaceImmediateSubElements(Type type,
3535
MLIRContext *Type::getContext() const { return getDialect().getContext(); }
3636

3737
bool Type::isFloat8E5M2() const { return llvm::isa<Float8E5M2Type>(*this); }
38+
bool Type::isFloat8E4M3() const { return llvm::isa<Float8E4M3Type>(*this); }
3839
bool Type::isFloat8E4M3FN() const { return llvm::isa<Float8E4M3FNType>(*this); }
3940
bool Type::isFloat8E5M2FNUZ() const {
4041
return llvm::isa<Float8E5M2FNUZType>(*this);

‎mlir/python/mlir/_mlir_libs/_mlir/ir.pyi

+14
Original file line numberDiff line numberDiff line change
@@ -123,6 +123,7 @@ __all__ = [
123123
"Float8E4M3B11FNUZType",
124124
"Float8E4M3FNType",
125125
"Float8E4M3FNUZType",
126+
"Float8E4M3Type",
126127
"Float8E5M2FNUZType",
127128
"Float8E5M2Type",
128129
"FloatAttr",
@@ -1575,6 +1576,19 @@ class Float8E4M3FNUZType(FloatType):
15751576
@property
15761577
def typeid(self) -> TypeID: ...
15771578

1579+
class Float8E4M3Type(FloatType):
1580+
static_typeid: ClassVar[TypeID]
1581+
@staticmethod
1582+
def get(context: Optional[Context] = None) -> Float8E4M3Type:
1583+
"""
1584+
Create a float8_e4m3 type.
1585+
"""
1586+
@staticmethod
1587+
def isinstance(other: Type) -> bool: ...
1588+
def __init__(self, cast_from_type: Type) -> None: ...
1589+
@property
1590+
def typeid(self) -> TypeID: ...
1591+
15781592
class Float8E5M2FNUZType(FloatType):
15791593
static_typeid: ClassVar[TypeID]
15801594
@staticmethod

‎mlir/python/mlir/extras/types.py

+2
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@
1414
F64Type,
1515
Float8E4M3B11FNUZType,
1616
Float8E4M3FNType,
17+
Float8E4M3Type,
1718
Float8E5M2Type,
1819
FunctionType,
1920
IndexType,
@@ -68,6 +69,7 @@ def ui(width):
6869
bf16 = lambda: BF16Type.get()
6970

7071
f8E5M2 = lambda: Float8E5M2Type.get()
72+
f8E4M3 = lambda: Float8E4M3Type.get()
7173
f8E4M3FN = lambda: Float8E4M3FNType.get()
7274
f8E4M3B11FNUZ = lambda: Float8E4M3B11FNUZType.get()
7375

‎mlir/test/IR/attribute.mlir

+4
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,10 @@ func.func @float_attrs_pass() {
4040
// CHECK: float_attr = 2.000000e+00 : f8E5M2
4141
float_attr = 2. : f8E5M2
4242
} : () -> ()
43+
"test.float_attrs"() {
44+
// CHECK: float_attr = 2.000000e+00 : f8E4M3
45+
float_attr = 2. : f8E4M3
46+
} : () -> ()
4347
"test.float_attrs"() {
4448
// CHECK: float_attr = 2.000000e+00 : f8E4M3FN
4549
float_attr = 2. : f8E4M3FN

‎mlir/test/Target/LLVMIR/llvmir.mlir

+3
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,9 @@ llvm.mlir.global internal constant @string_const("foobar") : !llvm.array<6 x i8>
3939
// CHECK: @int_global_undef = internal global i64 undef
4040
llvm.mlir.global internal @int_global_undef() : i64
4141

42+
// CHECK: @f8E4M3_global_as_i8 = internal global i8 60
43+
llvm.mlir.global internal @f8E4M3_global_as_i8(1.5 : f8E4M3) : i8
44+
4245
// CHECK: @f8E4M3FN_global_as_i8 = internal global i8 60
4346
llvm.mlir.global internal @f8E4M3FN_global_as_i8(1.5 : f8E4M3FN) : i8
4447

‎mlir/test/python/ir/builtin_types.py

+9
Original file line numberDiff line numberDiff line change
@@ -113,6 +113,8 @@ def testTypeIsInstance():
113113
def testFloatTypeSubclasses():
114114
ctx = Context()
115115
# CHECK: True
116+
print(isinstance(Type.parse("f8E4M3", ctx), FloatType))
117+
# CHECK: True
116118
print(isinstance(Type.parse("f8E4M3FN", ctx), FloatType))
117119
# CHECK: True
118120
print(isinstance(Type.parse("f8E5M2", ctx), FloatType))
@@ -229,6 +231,8 @@ def testIndexType():
229231
@run
230232
def testFloatType():
231233
with Context():
234+
# CHECK: float: f8E4M3
235+
print("float:", Float8E4M3Type.get())
232236
# CHECK: float: f8E4M3FN
233237
print("float:", Float8E4M3FNType.get())
234238
# CHECK: float: f8E5M2
@@ -601,6 +605,7 @@ def testTypeIDs():
601605
types = [
602606
(IntegerType, IntegerType.get_signless(16)),
603607
(IndexType, IndexType.get()),
608+
(Float8E4M3Type, Float8E4M3Type.get()),
604609
(Float8E4M3FNType, Float8E4M3FNType.get()),
605610
(Float8E5M2Type, Float8E5M2Type.get()),
606611
(Float8E4M3FNUZType, Float8E4M3FNUZType.get()),
@@ -624,6 +629,7 @@ def testTypeIDs():
624629

625630
# CHECK: IntegerType(i16)
626631
# CHECK: IndexType(index)
632+
# CHECK: Float8E4M3Type(f8E4M3)
627633
# CHECK: Float8E4M3FNType(f8E4M3FN)
628634
# CHECK: Float8E5M2Type(f8E5M2)
629635
# CHECK: Float8E4M3FNUZType(f8E4M3FNUZ)
@@ -704,6 +710,9 @@ def print_downcasted(typ):
704710
# CHECK: Float8E4M3B11FNUZType
705711
# CHECK: Float8E4M3B11FNUZType(f8E4M3B11FNUZ)
706712
print_downcasted(Float8E4M3B11FNUZType.get())
713+
# CHECK: Float8E4M3Type
714+
# CHECK: Float8E4M3Type(f8E4M3)
715+
print_downcasted(Float8E4M3Type.get())
707716
# CHECK: Float8E4M3FNType
708717
# CHECK: Float8E4M3FNType(f8E4M3FN)
709718
print_downcasted(Float8E4M3FNType.get())

0 commit comments

Comments
 (0)