Skip to content
Snippets Groups Projects
Commit 4037a90d authored by Benjamin Halimi's avatar Benjamin Halimi
Browse files

fix appendRoundClip() (associateInput/associateOutput)

parent dcb0db5e
No related branches found
No related tags found
2 merge requests!54Update 0.3.1 -> 0.4.0,!50Enhancement : Quantizer only PTQ
Pipeline #69620 failed
......@@ -119,7 +119,7 @@ void appendRoundClip(std::shared_ptr<Node> metaOpNode, double clipMin, double cl
auto newMulNode = Mul();
auto roundNode = Round();
auto clipNode = Clip(""); //, clipMin, clipMax);
auto newCoeffNode = mulNode->getParent(1)->clone(); // UUU Producer(coeffTensor);
auto newCoeffNode = mulNode->getParent(1)->clone(); // XXX Producer(coeffTensor);
// create the new micrograph
......@@ -130,9 +130,12 @@ void appendRoundClip(std::shared_ptr<Node> metaOpNode, double clipMin, double cl
// manually connect the IOs !!!
auto newMulOp = std::static_pointer_cast<OperatorTensor> (newMulNode->getOperator());
newMulOp->setInput(0, mulOp->getInput(0)); // MANDATORY (because we need an input tensor)
newMulOp->associateInput(0, mulOp->getInput(0)); // MANDATORY (because we need an input tensor)
auto clipOp = std::static_pointer_cast<Clip_Op> (clipNode->getOperator());
clipOp->setOutput(0, mulOp->getOutput(0)); // MANDATORY ? YES !!!
clipOp->associateOutput(0, mulOp->getOutput(0)); // MANDATORY ? YES !!!
// Log::notice( " old mul ref count : {}", mulOp->getOutput(0)->getImpl().use_count());
// Log::notice( " new mul ref count : {}", newMulOp->getOutput(0)->getImpl().use_count());
// Connect the clip min and max tensors
......@@ -145,16 +148,15 @@ void appendRoundClip(std::shared_ptr<Node> metaOpNode, double clipMin, double cl
newMicroGraph->add(minNode);
newMicroGraph->add(maxNode);
// set backend
// set the backend/datatype
newMicroGraph->setBackend(backend);
newMicroGraph->setDataType(dataType);
// reset the scheduling
SequentialScheduler scheduler(newMicroGraph);
scheduler.resetScheduling();
//scheduler.generateScheduling();
// set the micrograph
......
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