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

Removed initGrad

parent 1132cf3f
No related branches found
No related tags found
2 merge requests!73version 0.2.3,!70Multiple refactors
Pipeline #48523 failed
...@@ -416,7 +416,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward ...@@ -416,7 +416,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
auto myProd = Producer(inputTensor, "prod"); auto myProd = Producer(inputTensor, "prod");
myProd -> addChild(gv); myProd -> addChild(gv);
gv -> compile("cpu", DataType::Float32); gv -> compile("cpu", DataType::Float32);
compile_gradient(gv);
SequentialScheduler scheduler(gv); SequentialScheduler scheduler(gv);
scheduler.forward(); scheduler.forward();
auto outNode = gv->getOrderedOutputs()[0].first; auto outNode = gv->getOrderedOutputs()[0].first;
...@@ -432,7 +432,6 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward ...@@ -432,7 +432,6 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
{6.0f, 6.0f, 6.0f, 6.0f, 6.0f}, {6.0f, 6.0f, 6.0f, 6.0f, 6.0f},
{6.0f, 6.0f, 6.0f, 7.0f, 7.0f}, {6.0f, 6.0f, 6.0f, 7.0f, 7.0f},
{7.0f, 7.0f, 7.0f, 7.0f, 7.0f}}}}}); {7.0f, 7.0f, 7.0f, 7.0f, 7.0f}}}}});
predictedOutput->initGrad();
predictedOutput->setGrad(targetOutput); predictedOutput->setGrad(targetOutput);
REQUIRE_NOTHROW(scheduler.backward()); REQUIRE_NOTHROW(scheduler.backward());
} }
......
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