Skip to content
Snippets Groups Projects
Commit 4e1f36cb authored by Maxence Naud's avatar Maxence Naud
Browse files

[Upd] backward test in Test_Scheduler.cpp

parent 5d4b2967
No related branches found
No related tags found
2 merge requests!50version 0.2.0,!39Scheduler backprop
Pipeline #39202 failed
......@@ -210,7 +210,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
}
TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward]") {
std::shared_ptr<GraphView> gv = Sequential({ReLU(), ReLU()});
std::shared_ptr<GraphView> gv = Sequential({ReLU("relu0"), Sqrt("srqt0"), ReLU("relu1")});
std::shared_ptr<Tensor> inputTensor =
std::make_shared<Tensor>(Array4D<int, 2, 1, 5, 5>{{{{{0, 1, 2, 3, 4},
......@@ -227,7 +227,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
// implem already set to default
auto myProd = Producer(inputTensor, "prod");
myProd -> addChild(gv);
gv -> compile("cpu", DataType::Float32);
gv -> compile("cpu", DataType::Int32);
compile_gradient(gv);
SequentialScheduler scheduler(gv);
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