cleanup(compiler): Fix warning about comparison of expressions with different signedness in TensorOpsToLinalg

This commit is contained in:
Quentin Bourgerie
2021-11-09 16:09:50 +01:00
parent af0819c403
commit 3fad9870a8

View File

@@ -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));