From 3fad9870a855b75a97a2a101d4267b0cff5a0b03 Mon Sep 17 00:00:00 2001 From: Quentin Bourgerie Date: Tue, 9 Nov 2021 16:09:50 +0100 Subject: [PATCH] cleanup(compiler): Fix warning about comparison of expressions with different signedness in TensorOpsToLinalg --- .../Conversion/HLFHETensorOpsToLinalg/TensorOpsToLinalg.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/compiler/lib/Conversion/HLFHETensorOpsToLinalg/TensorOpsToLinalg.cpp b/compiler/lib/Conversion/HLFHETensorOpsToLinalg/TensorOpsToLinalg.cpp index 7b08adc0f..60734a59a 100644 --- a/compiler/lib/Conversion/HLFHETensorOpsToLinalg/TensorOpsToLinalg.cpp +++ b/compiler/lib/Conversion/HLFHETensorOpsToLinalg/TensorOpsToLinalg.cpp @@ -130,8 +130,8 @@ getBroadcastedAffineMap(const mlir::RankedTensorType &resultType, auto operandShape = operandType.getShape(); affineExprs.reserve(resultShape.size()); size_t deltaNumDim = resultShape.size() - operandShape.size(); - for (auto i = 0; i < operandShape.size(); i++) { - if (operandShape[i] == 1 && resultShape[i + deltaNumDim] != 1) { + for (size_t i = 0; i < operandShape.size(); i++) { + if (operandShape[i] == 1) { affineExprs.push_back(rewriter.getAffineConstantExpr(0)); } else { affineExprs.push_back(rewriter.getAffineDimExpr(i + deltaNumDim));