allow bf16 for tosa.avgpool2d in verifier

Xilinx · Feb 5, 2024 · 5e87840 · 5e87840
1 parent 845176a
commit 5e87840
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
@@ -193,6 +193,8 @@ LogicalResult tosa::AvgPool2dOp::verify() {
 
   if (inputETy.isF32() && resultETy.isF32())
     return success();
+  if (inputETy.isBF16() && resultETy.isBF16())
+    return success();
   if (inputETy.isInteger(8) && resultETy.isInteger(8))
     return success();
   if (inputETy.isInteger(16) && resultETy.isInteger(16))