[Mlir-commits] [mlir] [mlir][tosa] Add FP8 support (PR #127730)
Luke Hutton
llvmlistbot at llvm.org
Wed Feb 19 08:52:07 PST 2025
================
@@ -459,16 +472,123 @@ LogicalResult tosa::AvgPool2dOp::verify() {
if (inputETy.isF32() && !accType.isF32())
return emitOpError("accumulator type for f32 tensor is not f32");
+ if ((llvm::isa<Float8E5M2Type>(inputETy) ||
+ llvm::isa<Float8E4M3FNType>(inputETy)) &&
+ !accType.isF16())
+ return emitOpError("accumulator type for f8 tensor is not f16");
----------------
lhutton1 wrote:
(added a related comment above before I saw this one: https://github.com/llvm/llvm-project/pull/127730/files#r1962023091)
https://github.com/llvm/llvm-project/pull/127730
More information about the Mlir-commits
mailing list