llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix `uint64_t` type. (#3519) 

`u_int64_t` is nonstandard and does not exist in MSVC.
pull/3521/head
Scott Todd 2024-07-02 09:06:20 -07:00 committed by GitHub
parent f1e3701caf
commit ca0e906675
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp
View File

@ -2456,8 +2456,8 @@ public:
// Arragne reduce_dims tensor (vector), [0, 1, ... , dim-1, dim+1, ... , // Arragne reduce_dims tensor (vector), [0, 1, ... , dim-1, dim+1, ... ,
// ndim-1] // ndim-1]
llvm::SmallVector<Value> reduceDimsVector; llvm::SmallVector<Value> reduceDimsVector;
for (u_int64_t i = 0; i < ndim; i++) { for (uint64_t i = 0; i < ndim; i++) {
if (i == (u_int64_t)dimInt) if (i == (uint64_t)dimInt)
continue; continue;
Value constI = rewriter.create<Torch::ConstantIntOp>( Value constI = rewriter.create<Torch::ConstantIntOp>(
@ -2473,8 +2473,8 @@ public:
// Make output shape for linalg.vector_norm operation // Make output shape for linalg.vector_norm operation
SmallVector<Value> inputSizeValue; SmallVector<Value> inputSizeValue;
for (u_int64_t i = 0; i < inputSize.size(); i++) { for (uint64_t i = 0; i < inputSize.size(); i++) {
if (i != (u_int64_t)dimInt) if (i != (uint64_t)dimInt)
inputSize[i] = 1; inputSize[i] = 1;
inputSizeValue.push_back( inputSizeValue.push_back(