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

MergeNode ensure that channel must have 2 users
All checks were successful
Validate Operations / validate-operations (push) Successful in 16m7s
Details

Browse Source
This commit is contained in:
ilgeco
2026-04-09 10:41:49 +02:00
parent ece24867e4
commit b468858d25
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/PIM/Dialect/Spatial/Transforms/MergeComputeNode/MergeComputeNodePass.cpp
View File

@@ -61,11 +61,9 @@ public:
ChannelOrLocalOp getAsChannelValueAndInsertSender(SpatWeightedCompute spatWeightedCompute) { ChannelOrLocalOp getAsChannelValueAndInsertSender(SpatWeightedCompute spatWeightedCompute) {
if (channelSendInserter == nullptr) { auto [first, second] = channelNewInserter();
auto [first, second] = channelNewInserter(); channelNewOpVal = first;
channelNewOpVal = first; channelSendInserter = second;
channelSendInserter = second;
}
auto BB = computeResults.innerValue.getParentBlock(); auto BB = computeResults.innerValue.getParentBlock();
if (!BB->empty() && isa<spatial::SpatYieldOp>(BB->back())) if (!BB->empty() && isa<spatial::SpatYieldOp>(BB->back()))
insertPointSend = InsertPoint(BB, --BB->end()); insertPointSend = InsertPoint(BB, --BB->end());
@@ -292,7 +290,8 @@ private:
} }
else { else {
auto newInst = rewriter.clone(op, mapper); auto newInst = rewriter.clone(op, mapper);
//TODO Refactor in a lambda? same code just different cast, but templated lambda are C++20 and a free function is a bit too much // TODO Refactor in a lambda? same code just different cast, but templated lambda are C++20 and a free function
// is a bit too much
if (auto vmOp = llvm::dyn_cast<spatial::SpatWeightedMVMOp>(newInst)) { if (auto vmOp = llvm::dyn_cast<spatial::SpatWeightedMVMOp>(newInst)) {
auto oldIndex = vmOp.getWeightIndex(); auto oldIndex = vmOp.getWeightIndex();
auto newWeight = mapper.lookup(*std::next(fromCompute.getWeights().begin(), oldIndex)); auto newWeight = mapper.lookup(*std::next(fromCompute.getWeights().begin(), oldIndex));