Commit d84d1557 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon

CPU: Enable tuple result test (missed in the previous commit)

parent 4fa2d1a1
...@@ -202,9 +202,8 @@ TEST(cpu, abc_tuple_int64) ...@@ -202,9 +202,8 @@ TEST(cpu, abc_tuple_int64)
ASSERT_EQ((vector<element::Int64::type>{50, 72, 98, 128}), result->get_vector()); ASSERT_EQ((vector<element::Int64::type>{50, 72, 98, 128}), result->get_vector());
} }
/*
// Multiple retrive values // Multiple retrive values
TEST(execute, tuple_result) TEST(cpu, tuple_result)
{ {
auto shape = Shape{2, 2}; auto shape = Shape{2, 2};
auto A = make_shared<op::Parameter>(element::Float32::element_type(), shape); auto A = make_shared<op::Parameter>(element::Float32::element_type(), shape);
...@@ -219,7 +218,7 @@ TEST(execute, tuple_result) ...@@ -219,7 +218,7 @@ TEST(execute, tuple_result)
auto f = make_shared<Function>( auto f = make_shared<Function>(
make_shared<op::Tuple>(Nodes{A_add_B, A_add_B_mul_C}), rt, op::Parameters{A, B, C}); make_shared<op::Tuple>(Nodes{A_add_B, A_add_B_mul_C}), rt, op::Parameters{A, B, C});
auto manager = runtime::Manager::get("NGVM"); auto manager = runtime::Manager::get("CPU");
auto external = manager->compile(f); auto external = manager->compile(f);
auto backend = manager->allocate_backend(); auto backend = manager->allocate_backend();
auto cf = backend->make_call_frame(external); auto cf = backend->make_call_frame(external);
...@@ -241,6 +240,7 @@ TEST(execute, tuple_result) ...@@ -241,6 +240,7 @@ TEST(execute, tuple_result)
ASSERT_EQ((vector<float>{54, 80, 110, 144}), r1->get_vector()); ASSERT_EQ((vector<float>{54, 80, 110, 144}), r1->get_vector());
} }
/*
TEST(execute, abs) TEST(execute, abs)
{ {
auto shape = Shape{2, 2}; auto shape = Shape{2, 2};
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment