[Mlir-commits] [mlir] [MLIR][AMDGPU] Add support for fp8 ops on gfx12 (PR #106388)
Giuseppe Rossini
llvmlistbot at llvm.org
Mon Sep 2 12:04:08 PDT 2024
================
@@ -235,7 +235,9 @@ LogicalResult WMMAOp::verify() {
bool isDestFloat =
(destElemType.isF32() || destElemType.isF16() || destElemType.isBF16());
- bool isSrcFloat = (sourceAElemType.isF16() || sourceAElemType.isBF16());
+ bool isSrcFloat =
+ (sourceAElemType.isF16() || sourceAElemType.isBF16() ||
+ sourceAElemType.isFloat8E4M3FN() || sourceAElemType.isFloat8E5M2());
----------------
giuseros wrote:
Done
https://github.com/llvm/llvm-project/pull/106388
More information about the Mlir-commits
mailing list