Skip to content

N
ngraph
Commit e47cebea authored by Geoffrey Wenger's avatar Geoffrey Wenger Committed by Sang Ik Lee
Browse files
Options

Fixed gather testing (#4007)

parent ee8ddabd
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 19 deletions
test/backend/gather.in.cpp
View file @ e47cebea
...@@ -625,8 +625,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int8) ...@@ -625,8 +625,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int8)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<int8_t>{10, 11, 20, 21, 20, 21, 30, 31}), EXPECT_TRUE(test::all_close((vector<int8_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<int8_t>(result), read_vector<int8_t>(result)));
static_cast<int8_t> MIN_FLOAT_TOLERANCE_BITS));
} }
NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int16) NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int16)
...@@ -651,8 +650,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int16) ...@@ -651,8 +650,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int16)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<int16_t>{10, 11, 20, 21, 20, 21, 30, 31}), EXPECT_TRUE(test::all_close((vector<int16_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<int16_t>(result), read_vector<int16_t>(result)));
static_cast<int16_t> MIN_FLOAT_TOLERANCE_BITS));
} }
NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int32) NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int32)
...@@ -677,8 +675,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int32) ...@@ -677,8 +675,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int32)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<int32_t>{10, 11, 20, 21, 20, 21, 30, 31}), EXPECT_TRUE(test::all_close((vector<int32_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<int32_t>(result), read_vector<int32_t>(result)));
MIN_FLOAT_TOLERANCE_BITS));
} }
NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int64) NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int64)
...@@ -703,8 +700,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int64) ...@@ -703,8 +700,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int64)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<int64_t>{10, 11, 20, 21, 20, 21, 30, 31}), EXPECT_TRUE(test::all_close((vector<int64_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<int64_t>(result), read_vector<int64_t>(result)));
static_cast<int64_t> MIN_FLOAT_TOLERANCE_BITS));
} }
NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint8) NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint8)
...@@ -729,8 +725,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint8) ...@@ -729,8 +725,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint8)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<uint8_t>{10, 11, 20, 21, 20, 21, 30, 31}), EXPECT_TRUE(test::all_close((vector<uint8_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<uint8_t>(result), read_vector<uint8_t>(result)));
static_cast<uint8_t> MIN_FLOAT_TOLERANCE_BITS));
} }
NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint16) NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint16)
...@@ -755,8 +750,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint16) ...@@ -755,8 +750,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint16)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<uint16_t>{10, 11, 20, 21, 20, 21, 30, 31}), EXPECT_TRUE(test::all_close((vector<uint16_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<uint16_t>(result), read_vector<uint16_t>(result)));
static_cast<uint16_t> MIN_FLOAT_TOLERANCE_BITS));
} }
NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint32) NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint32)
...@@ -781,8 +775,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint32) ...@@ -781,8 +775,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint32)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<uint32_t>{10, 11, 20, 21, 20, 21, 30, 31}), EXPECT_TRUE(test::all_close((vector<uint32_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<uint32_t>(result), read_vector<uint32_t>(result)));
static_cast<uint32_t> MIN_FLOAT_TOLERANCE_BITS));
} }
NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint64) NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint64)
...@@ -807,8 +800,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint64) ...@@ -807,8 +800,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint64)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<uint64_t>{10, 11, 20, 21, 20, 21, 30, 31}), EXPECT_TRUE(test::all_close((vector<uint64_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<uint64_t>(result), read_vector<uint64_t>(result)));
static_cast<uint64_t> MIN_FLOAT_TOLERANCE_BITS));
} }
NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_bool) NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_bool)
...@@ -832,7 +824,5 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_bool) ...@@ -832,7 +824,5 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_bool)
auto c = backend->compile(f); auto c = backend->compile(f);
c->call_with_validate({result}, {p, i}); c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<char>{1, 1, 1, 0, 1, 0, 0, 1}), EXPECT_TRUE(test::all_close((vector<char>{1, 1, 1, 0, 1, 0, 0, 1}), read_vector<char>(result)));
read_vector<char>(result),
static_cast<char> MIN_FLOAT_TOLERANCE_BITS));
} }
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