Skip to content

Commit

Permalink
Fix uint64_t type. (#3519)
Browse files Browse the repository at this point in the history
`u_int64_t` is nonstandard and does not exist in MSVC.
  • Loading branch information
ScottTodd committed Jul 2, 2024
1 parent f1e3701 commit ca0e906
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -2456,8 +2456,8 @@ class DecomposeAtenRenormOp : public OpRewritePattern<AtenRenormOp> {
// Arragne reduce_dims tensor (vector), [0, 1, ... , dim-1, dim+1, ... ,
// ndim-1]
llvm::SmallVector<Value> 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<Torch::ConstantIntOp>(
Expand All @@ -2473,8 +2473,8 @@ class DecomposeAtenRenormOp : public OpRewritePattern<AtenRenormOp> {

// Make output shape for linalg.vector_norm operation
SmallVector<Value> 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(
Expand Down

0 comments on commit ca0e906

Please sign in to comment.