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
This commit is part of merge request !39. Comments created here will be created in the context of that merge request.
...@@ -349,7 +349,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") { ...@@ -349,7 +349,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(forward)") {
} }
TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward]") { 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::shared_ptr<Tensor> inputTensor =
std::make_shared<Tensor>(Array4D<int, 2, 1, 5, 5>{{{{{0, 1, 2, 3, 4}, 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 ...@@ -366,7 +366,7 @@ TEST_CASE("[cpu/scheduler] SequentialScheduler(backward)", "[scheduler][backward
// implem already set to default // implem already set to default
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::Int32);
compile_gradient(gv); compile_gradient(gv);
SequentialScheduler scheduler(gv); SequentialScheduler scheduler(gv);
scheduler.backward(); 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