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 ...@@ -150,7 +150,7 @@ void Aidge::Memorize_Op::setBackend(const std::string& name, Aidge::DeviceIdx_t
} }
void Aidge::Memorize_Op::forward() { void Aidge::Memorize_Op::forward() {
Operator::forward(); OperatorTensor::forward();
++mAttributes->template getAttr<MemorizeAttr::ForwardStep>(); ++mAttributes->template getAttr<MemorizeAttr::ForwardStep>();
mAttributes->template getAttr<MemorizeAttr::ScheduleStep>() = 0; mAttributes->template getAttr<MemorizeAttr::ScheduleStep>() = 0;
} }
......
...@@ -24,6 +24,7 @@ Aidge::Elts_t Aidge::Pop_ProdConso::getNbRequiredData(const Aidge::IOIndex_t inp ...@@ -24,6 +24,7 @@ Aidge::Elts_t Aidge::Pop_ProdConso::getNbRequiredData(const Aidge::IOIndex_t inp
assert(mOp.getRawInput(inputIdx) && "requires valid input"); assert(mOp.getRawInput(inputIdx) && "requires valid input");
const Pop_Op& op = dynamic_cast<const Pop_Op&>(mOp); 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() return Elts_t::DataElts(op.getInput(inputIdx)->size()
/ op.getInput(inputIdx)->dims()[0]); / op.getInput(inputIdx)->dims()[0]);
} }
...@@ -93,7 +94,7 @@ std::set<std::string> Aidge::Pop_Op::getAvailableBackends() const { ...@@ -93,7 +94,7 @@ std::set<std::string> Aidge::Pop_Op::getAvailableBackends() const {
} }
void Aidge::Pop_Op::forward() { void Aidge::Pop_Op::forward() {
Operator::forward(); OperatorTensor::forward();
++mAttributes->template getAttr<PopAttr::ForwardStep>(); ++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