From 3254742b6336492631e6dd1c22e3bd6bd862fe4d Mon Sep 17 00:00:00 2001 From: Jerry Ge Date: Mon, 7 Apr 2025 13:47:10 -0700 Subject: [PATCH] [mlir][tosa] Add accumulator verifier for FP8 types for AvgPool2D Signed-off-by: Jerry Ge Change-Id: I04b34f62e622893b52417a96ab33166c96c846d7 --- mlir/lib/Dialect/Tosa/IR/TosaOps.cpp | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp index c8e9ad8bd3346..1b73158cd7041 100644 --- a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp +++ b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp @@ -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(inputETy) || + llvm::isa(inputETy)) && + !accType.isF16()) + return emitOpError("accumulator type for f8 tensor is not f16"); + if (inputETy != inputZpETy) return emitOpError("expect both input and its zero point are the same " "element type, got ")