[locomotiv] Use link helper (#6197)
author박종현/On-Device Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Mon, 5 Aug 2019 06:05:35 +0000 (15:05 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Mon, 5 Aug 2019 06:05:35 +0000 (15:05 +0900)
Let's use link helper instead of deprecated "node" method in GraphInput/Output.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>
compiler/locomotiv/src/Session.test.cpp

index 27fe074..07d55d2 100644 (file)
@@ -41,7 +41,7 @@ TEST(Session, graph_IO_size)
   for (uint32_t i = 0; i < inputs; ++i)
   {
     auto pull = g->nodes()->create<loco::Pull>();
-    g->inputs()->create()->node(pull);
+    loco::link(g->inputs()->create(), pull);
   }
 
   // outputs
@@ -49,7 +49,7 @@ TEST(Session, graph_IO_size)
   for (uint32_t o = 0; o < outputs; ++o)
   {
     auto push = g->nodes()->create<loco::Push>();
-    g->outputs()->create()->node(push);
+    loco::link(g->outputs()->create(), push);
   }
 
   // Make session
@@ -67,7 +67,7 @@ TEST(Session, set_input)
   pull->dtype(loco::DataType::FLOAT32);
   pull->rank(1);
   pull->dim(0) = 1;
-  g->inputs()->create()->node(pull);
+  loco::link(g->inputs()->create(), pull);
 
   // Make good data
   auto buf = make_buffer<float, LexicalLayout>(Shape{1});
@@ -115,11 +115,11 @@ TEST(Session, inference_identity)
 
     // Input
     auto graph_input = g->inputs()->create();
-    graph_input->node(pull_node);
+    loco::link(graph_input, pull_node);
 
     // Output
     auto graph_output = g->outputs()->create();
-    graph_output->node(push_node);
+    loco::link(graph_output, push_node);
 
     graphs.push_back(std::move(g));
   }
@@ -140,11 +140,11 @@ TEST(Session, inference_identity)
 
     // Input
     auto graph_input = g->inputs()->create();
-    graph_input->node(pull_node);
+    loco::link(graph_input, pull_node);
 
     // Output
     auto graph_output = g->outputs()->create();
-    graph_output->node(push_node);
+    loco::link(graph_output, push_node);
 
     graphs.push_back(std::move(g));
   }
@@ -350,7 +350,7 @@ TEST(Session, dtor)
 
   // Input
   auto input = g->inputs()->create();
-  input->node(pull);
+  loco::link(input, pull);
 
   {
     locomotiv::Session s(g.get());