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

fix build error

pull/3643/head
Rob Suderman 2024-08-16 16:17:31 -07:00
parent d23ecab1ef
commit 3697b0f3b6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Conversion/TorchOnnxToTorch/DefaultDomainAtoF.cpp
View File

@ -175,7 +175,7 @@ void mlir::torch::onnx_c::populateDefaultDomainAtoF(
SmallVector<Value> accSqGrads; SmallVector<Value> accSqGrads;
size_t inputCount = (operands.size() - 2) / 4; size_t inputCount = (operands.size() - 2) / 4;
for (int i = 0; i < inputCount; ++i) { for (size_t i = 0; i < inputCount; ++i) {
inputs.push_back(operands[2 + i]); inputs.push_back(operands[2 + i]);
grads.push_back(operands[2 + i + inputCount * 1]); grads.push_back(operands[2 + i + inputCount * 1]);
accGrads.push_back(operands[2 + i + inputCount * 2]); accGrads.push_back(operands[2 + i + inputCount * 2]);