Skip to content
Snippets Groups Projects
Commit 0eaea479 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Merge branch 'fix_lstm2' into 'dev'

Partial fix for issue eclipse/aidge/aidge_core#167

See merge request !252
parents 1fda3d80 2261a76b
No related branches found
No related tags found
2 merge requests!279v0.4.0,!252Partial fix for issue eclipse/aidge/aidge_core#167
Pipeline #59529 passed
......@@ -150,7 +150,7 @@ void Aidge::Memorize_Op::setBackend(const std::string& name, Aidge::DeviceIdx_t
}
void Aidge::Memorize_Op::forward() {
Operator::forward();
OperatorTensor::forward();
++mAttributes->template getAttr<MemorizeAttr::ForwardStep>();
mAttributes->template getAttr<MemorizeAttr::ScheduleStep>() = 0;
}
......
......@@ -24,6 +24,7 @@ Aidge::Elts_t Aidge::Pop_ProdConso::getNbRequiredData(const Aidge::IOIndex_t inp
assert(mOp.getRawInput(inputIdx) && "requires valid input");
const Pop_Op& op = dynamic_cast<const Pop_Op&>(mOp);
AIDGE_ASSERT(!op.getInput(inputIdx)->empty(), "Pop operator requires known, non-empty, input dims for scheduling. You might have an unresolved data dependency upstream in the computing graph.");
return Elts_t::DataElts(op.getInput(inputIdx)->size()
/ op.getInput(inputIdx)->dims()[0]);
}
......@@ -93,7 +94,7 @@ std::set<std::string> Aidge::Pop_Op::getAvailableBackends() const {
}
void Aidge::Pop_Op::forward() {
Operator::forward();
OperatorTensor::forward();
++mAttributes->template getAttr<PopAttr::ForwardStep>();
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment