Skip to content
This repository has been archived by the owner on Jan 3, 2023. It is now read-only.

Commit

Permalink
Fixed gather testing (#4007)
Browse files Browse the repository at this point in the history
  • Loading branch information
@silee2
Geoffrey Wenger authored and silee2 committed Dec 6, 2019
1 parent ee8ddab commit e47cebe
Showing 1 changed file with 9 additions and 19 deletions.
28 changes: 9 additions & 19 deletions test/backend/gather.in.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -625,8 +625,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int8)
auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<int8_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<int8_t>(result),
static_cast<int8_t> MIN_FLOAT_TOLERANCE_BITS));
read_vector<int8_t>(result)));
}

NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int16)
Expand All @@ -651,8 +650,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int16)
auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<int16_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<int16_t>(result),
static_cast<int16_t> MIN_FLOAT_TOLERANCE_BITS));
read_vector<int16_t>(result)));
}

NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int32)
Expand All @@ -677,8 +675,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int32)
auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<int32_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<int32_t>(result),
MIN_FLOAT_TOLERANCE_BITS));
read_vector<int32_t>(result)));
}

NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int64)
Expand All @@ -703,8 +700,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_int64)
auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<int64_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<int64_t>(result),
static_cast<int64_t> MIN_FLOAT_TOLERANCE_BITS));
read_vector<int64_t>(result)));
}

NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint8)
Expand All @@ -729,8 +725,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint8)
auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<uint8_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<uint8_t>(result),
static_cast<uint8_t> MIN_FLOAT_TOLERANCE_BITS));
read_vector<uint8_t>(result)));
}

NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint16)
Expand All @@ -755,8 +750,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint16)
auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<uint16_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<uint16_t>(result),
static_cast<uint16_t> MIN_FLOAT_TOLERANCE_BITS));
read_vector<uint16_t>(result)));
}

NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint32)
Expand All @@ -781,8 +775,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint32)
auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<uint32_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<uint32_t>(result),
static_cast<uint32_t> MIN_FLOAT_TOLERANCE_BITS));
read_vector<uint32_t>(result)));
}

NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint64)
Expand All @@ -807,8 +800,7 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_uint64)
auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<uint64_t>{10, 11, 20, 21, 20, 21, 30, 31}),
read_vector<uint64_t>(result),
static_cast<uint64_t> MIN_FLOAT_TOLERANCE_BITS));
read_vector<uint64_t>(result)));
}

NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_bool)
Expand All @@ -832,7 +824,5 @@ NGRAPH_TEST(${BACKEND_NAME}, gather_no_axis_bool)

auto c = backend->compile(f);
c->call_with_validate({result}, {p, i});
EXPECT_TRUE(test::all_close((vector<char>{1, 1, 1, 0, 1, 0, 0, 1}),
read_vector<char>(result),
static_cast<char> MIN_FLOAT_TOLERANCE_BITS));
EXPECT_TRUE(test::all_close((vector<char>{1, 1, 1, 0, 1, 0, 0, 1}), read_vector<char>(result)));
}

0 comments on commit e47cebe

Please sign in to comment.