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

Fix getAdaptation()

parent a0e58f13
No related branches found
No related tags found
2 merge requests!318[Upd] release verision 0.5.0,!302Fix getAdaptation()
Pipeline #63118 passed
......@@ -235,6 +235,11 @@ bool Aidge::OperatorImpl::checkIOSpec(const ImplSpec::IOSpec& required, const Im
}
std::shared_ptr<Aidge::Node> Aidge::OperatorImpl::getAdaptation(const ImplSpec& spec, const ImplSpec& requiredSpecs) const {
// Original graph is:
// --> {required IO specs} [node] {required IO specs} -->
// Obtained meta-op is:
// --> {required IO specs} [adapt inputs] --> {IO specs} [node] {IO specs} --> [adapt outputs] {required IO specs}
auto op = std::static_pointer_cast<OperatorTensor>(mOp.clone());
auto node = std::make_shared<Node>(op);
......@@ -306,7 +311,7 @@ std::shared_ptr<Aidge::Node> Aidge::OperatorImpl::getAdaptation(const ImplSpec&
cast->getOperator()->setBackend(node->getOperator()->backend());
parent->addChild(cast, i, 0);
op->getOutput(i)->setDataType(IOSpec.type);
op->getInput(i)->setDataType(IOSpec.type);
}
// Output format
......@@ -321,7 +326,7 @@ std::shared_ptr<Aidge::Node> Aidge::OperatorImpl::getAdaptation(const ImplSpec&
transposeOp->getOperator()->setBackend(node->getOperator()->backend());
parent->addChild(transposeOp, i, 0);
op->getOutput(i)->setDataFormat(IOSpec.format);
op->getInput(i)->setDataFormat(IOSpec.format);
}
// Output dims
......
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