Commit 199ec73e authored by Robert Kimball's avatar Robert Kimball Committed by Scott Cyphers

move unit test to correct file (#2685)

parent 846095fd
...@@ -95,3 +95,42 @@ NGRAPH_TEST(${BACKEND_NAME}, tensor_copy_from) ...@@ -95,3 +95,42 @@ NGRAPH_TEST(${BACKEND_NAME}, tensor_copy_from)
a->copy_from(*b); a->copy_from(*b);
EXPECT_TRUE(test::all_close(bv, read_vector<float>(a), 0.0f, 0.0f)); EXPECT_TRUE(test::all_close(bv, read_vector<float>(a), 0.0f, 0.0f));
} }
NGRAPH_TEST(${BACKEND_NAME}, get_parameters_and_results)
{
Shape shape{2, 2};
auto A = make_shared<op::Parameter>(element::f32, shape);
auto B = make_shared<op::Parameter>(element::f32, shape);
auto C = make_shared<op::Parameter>(element::f32, shape);
auto f = make_shared<Function>((A + B) * C, ParameterVector{A, B, C});
auto backend = runtime::Backend::create("${BACKEND_NAME}");
// Create some tensors for input/output
shared_ptr<runtime::Tensor> a = backend->create_tensor(element::f32, shape);
shared_ptr<runtime::Tensor> b = backend->create_tensor(element::f32, shape);
shared_ptr<runtime::Tensor> c = backend->create_tensor(element::f32, shape);
shared_ptr<runtime::Tensor> result = backend->create_tensor(element::f32, shape);
copy_data(a, test::NDArray<float, 2>({{1, 2}, {3, 4}}).get_vector());
copy_data(b, test::NDArray<float, 2>({{5, 6}, {7, 8}}).get_vector());
copy_data(c, test::NDArray<float, 2>({{9, 10}, {11, 12}}).get_vector());
auto handle = backend->compile(f);
auto parameters = handle->get_parameters();
auto results = handle->get_results();
ASSERT_EQ(parameters.size(), 3);
ASSERT_EQ(results.size(), 1);
// This part can't be enabled until we force backends to make a copy of the source graph
// auto func_parameters = f->get_parameters();
// auto func_results = f->get_results();
// for (size_t i = 0; i < 3; ++i)
// {
// EXPECT_NE(parameters[i], func_parameters[i]);
// }
// for (size_t i = 0; i < 1; ++i)
// {
// EXPECT_NE(results[i], func_results[i]);
// }
}
...@@ -7326,30 +7326,3 @@ NGRAPH_TEST(${BACKEND_NAME}, quantize_dynamic_offset) ...@@ -7326,30 +7326,3 @@ NGRAPH_TEST(${BACKEND_NAME}, quantize_dynamic_offset)
EXPECT_EQ((vector<output_c_type>{1, 1, 2, 3, 3, 3, 4, 5, 5, 5, 6, 7}), EXPECT_EQ((vector<output_c_type>{1, 1, 2, 3, 3, 3, 4, 5, 5, 5, 6, 7}),
read_vector<output_c_type>(y)); read_vector<output_c_type>(y));
} }
NGRAPH_TEST(${BACKEND_NAME}, get_parameters_and_results)
{
Shape shape{2, 2};
auto A = make_shared<op::Parameter>(element::f32, shape);
auto B = make_shared<op::Parameter>(element::f32, shape);
auto C = make_shared<op::Parameter>(element::f32, shape);
auto f = make_shared<Function>((A + B) * C, ParameterVector{A, B, C});
auto backend = runtime::Backend::create("${BACKEND_NAME}");
// Create some tensors for input/output
shared_ptr<runtime::Tensor> a = backend->create_tensor(element::f32, shape);
shared_ptr<runtime::Tensor> b = backend->create_tensor(element::f32, shape);
shared_ptr<runtime::Tensor> c = backend->create_tensor(element::f32, shape);
shared_ptr<runtime::Tensor> result = backend->create_tensor(element::f32, shape);
copy_data(a, test::NDArray<float, 2>({{1, 2}, {3, 4}}).get_vector());
copy_data(b, test::NDArray<float, 2>({{5, 6}, {7, 8}}).get_vector());
copy_data(c, test::NDArray<float, 2>({{9, 10}, {11, 12}}).get_vector());
auto handle = backend->compile(f);
auto parameters = handle->get_parameters();
auto results = handle->get_results();
EXPECT_EQ(parameters.size(), 3);
EXPECT_EQ(results.size(), 1);
}
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