Skip to content

Commit a1de6ab

Browse files
committed
Add two 8bit float types F8E4M3FN and F8E5M2 to valid XeGPU float type.
1 parent 435dbba commit a1de6ab

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mlir/include/mlir/Dialect/XeGPU/IR/XeGPUTypes.td

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ include "mlir/Dialect/XeGPU/IR/XeGPUDialect.td"
1414
include "mlir/IR/BuiltinTypes.td"
1515

1616
def XeGPU_IntType: AnyTypeOf<[I1, I8, I16, I32, I64, SI1, SI8, SI16, SI32, SI64, UI1, UI8, UI16, UI32, UI64]>;
17-
def XeGPU_FloatType: AnyTypeOf<[F16, F32, F64, BF16, TF32]>;
17+
def XeGPU_FloatType: AnyTypeOf<[F8E4M3FN, F8E5M2, F16, F32, F64, BF16, TF32]>;
1818
def XeGPU_ScalarType: AnyTypeOf<[XeGPU_IntType, XeGPU_FloatType]>;
1919
def XeGPU_PointerType : AnyTypeOf<[UI64, UI32, I64, I32]>;
2020
def XeGPU_BaseAddrType

0 commit comments

Comments
 (0)