[Mlir-commits] [mlir] [MLIR][AMDGPU] Add OCP FP8 support for new hardware (PR #106160)
Jakub Kuderski
llvmlistbot at llvm.org
Wed Sep 18 14:57:49 PDT 2024
================
@@ -454,6 +454,20 @@ static void wmmaPushOutputOperand(ConversionPatternRewriter &rewriter,
}
}
+/// Return true if `type` is the E5M2 variant of an 8-bit float that is
+/// supported by the `_bf8` instructions on the given `chipset`.
+static bool isNativeBf8(Chipset chipset, Type type) {
+ return (isGfx940Series(chipset) && type.isFloat8E5M2FNUZ()) ||
+ (hasOcpFp8(chipset) && type.isFloat8E5M2());
+}
+
+/// Return true if `type` is the E4M3FN variant of an 8-bit float that is
+/// supported by the `_fp8` instructions on the given `chipset`.
+static bool isNativeFp8(Chipset chipset, Type type) {
----------------
kuhar wrote:
```suggestion
static bool hasNativeFp8(Chipset chipset, Type type) {
```
https://github.com/llvm/llvm-project/pull/106160
More information about the Mlir-commits
mailing list