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

[Upd] backward test in Test_Scheduler.cpp

parent 86ecc9a5
No related branches found
No related tags found
2 merge requests!50version 0.2.0,!39Scheduler backprop
......@@ -349,7 +349,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},
......@@ -366,7 +366,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