diff --git a/mlir/include/mlir/Dialect/Tosa/Utils/ConversionUtils.h b/mlir/include/mlir/Dialect/Tosa/Utils/ConversionUtils.h index 88c2162928652..4e2f1b9cb19a9 100644 --- a/mlir/include/mlir/Dialect/Tosa/Utils/ConversionUtils.h +++ b/mlir/include/mlir/Dialect/Tosa/Utils/ConversionUtils.h @@ -145,7 +145,7 @@ TosaOp createOpAndInferShape(ImplicitLocOpBuilder &builder, Type resultTy, template TosaOp CreateOpAndInferShape(ImplicitLocOpBuilder &builder, Type resultTy, Args &&...args) { - if (TosaOp::template hasTrait()) { + if (TosaOp::template hasTrait<::mlir::OpTrait::SameOperandsAndResultRank>()) { // op requires same ranks for tensor operands if constexpr (sizeof...(Args) == 2) { auto argX = std::get<0>(std::tie(args...));