diff --git a/lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp b/lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp index f966b320c13..24a79cb0d31 100644 --- a/lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp +++ b/lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp @@ -2456,8 +2456,8 @@ class DecomposeAtenRenormOp : public OpRewritePattern { // Arragne reduce_dims tensor (vector), [0, 1, ... , dim-1, dim+1, ... , // ndim-1] llvm::SmallVector reduceDimsVector; - for (u_int64_t i = 0; i < ndim; i++) { - if (i == (u_int64_t)dimInt) + for (uint64_t i = 0; i < ndim; i++) { + if (i == (uint64_t)dimInt) continue; Value constI = rewriter.create( @@ -2473,8 +2473,8 @@ class DecomposeAtenRenormOp : public OpRewritePattern { // Make output shape for linalg.vector_norm operation SmallVector inputSizeValue; - for (u_int64_t i = 0; i < inputSize.size(); i++) { - if (i != (u_int64_t)dimInt) + for (uint64_t i = 0; i < inputSize.size(); i++) { + if (i != (uint64_t)dimInt) inputSize[i] = 1; inputSizeValue.push_back(