[Mlir-commits] [mlir] [mlir][tosa] Add accumulator verifier for FP8 types for AvgPool2D (PR #134727)
Luke Hutton
llvmlistbot at llvm.org
Tue Apr 8 02:03:41 PDT 2025
================
@@ -600,6 +600,11 @@ LogicalResult tosa::AvgPool2dOp::verify() {
if (inputETy.isF32() && !accType.isF32())
return emitOpError("accumulator type for f32 tensor is not f32");
+ if ((llvm::isa<Float8E5M2Type>(inputETy) ||
----------------
lhutton1 wrote:
+1 from me on this - it allows other users of TOSA to express operations that may be important for their workflow e.g. an fp8 avg_pool with an fp32 accumulator
https://github.com/llvm/llvm-project/pull/134727
More information about the Mlir-commits
mailing list