nnicolosi/Raptor
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

Fix add of input in random order for compute_batch
Validate Operations / validate-operations (push) Waiting to run
Details

Browse Source
This commit is contained in:
ilgeco
2026-05-22 15:21:02 +02:00
parent 074eb183c7
commit d136136d22
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/PIM/Dialect/Spatial/Transforms/MergeComputeNodes/MaterializeMergeSchedule.cpp
+11 -7
View File
@@ -551,15 +551,19 @@ BlockArgument appendInput(MaterializerState& state, MaterializedClass& materiali
return it->second; return it->second;
materializedClass.inputs.push_back(input); materializedClass.inputs.push_back(input);
if (auto compute = dyn_cast<SpatCompute>(materializedClass.op)) if (auto compute = dyn_cast<SpatCompute>(materializedClass.op)) {
compute.getInputsMutable().append(ValueRange(input)); compute.getInputsMutable().append(ValueRange(input));
else BlockArgument arg = materializedClass.body->addArgument(input.getType(), input.getLoc());
materializedClass.inputArgs[input] = arg;
}
else {
cast<SpatComputeBatch>(materializedClass.op).getInputsMutable().append(ValueRange(input)); cast<SpatComputeBatch>(materializedClass.op).getInputsMutable().append(ValueRange(input));
setOperandSegmentSizes(materializedClass.op, materializedClass.weights.size(), materializedClass.inputs.size()); setOperandSegmentSizes(materializedClass.op, materializedClass.weights.size(), materializedClass.inputs.size());
BlockArgument arg = materializedClass.body->insertArgument(materializedClass.body->getNumArguments()-1, input.getType(), input.getLoc());
BlockArgument arg = materializedClass.body->addArgument(input.getType(), input.getLoc()); materializedClass.inputArgs[input] = arg;
materializedClass.inputArgs[input] = arg; return arg;
return arg; }
llvm_unreachable("Cannot reach here");
} }
Value createIndexConstant(MaterializerState& state, Operation* anchor, int64_t value) { Value createIndexConstant(MaterializerState& state, Operation* anchor, int64_t value) {