Skip to content

GitLab

Commit f5329965 authored by Umang Yadav's avatar Umang Yadav
Browse files

Fix threshold for the dot op tests

parent 4487d4c9
Show whitespace changes
Inline Side-by-side
Showing with 118 additions and 115 deletions
test/ref/dot_op.cpp
View file @ f5329965
......@@ -41,7 +41,7 @@ void dot_2d_test()
1.33628943, -0.11413, -0.31270559, 1.59336732, -0.19361027,
-0.91620867, 0.40108416, -0.06969921, 0.68483471, -0.39906632,
-1.66423624, 0.69040076, -1.31490171, -0.11282616, -0.79391814};
std::vector<float> b = {6.09568541e-01,
std::vector<T> b = {6.09568541e-01,
-6.10527007e-01,
3.66646462e-01,
1.18951101e-01,
......@@ -66,7 +66,7 @@ void dot_2d_test()
auto result = p.eval({}).back();
std::vector<T> results_vector;
result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
std::vector<float> gold = {-1.56327541e+00,
std::vector<T> gold = {-1.56327541e+00,
-7.09570140e-01,
-5.37424982e-01,
-2.22994831e-01,
......@@ -78,7 +78,8 @@ void dot_2d_test()
-1.29885596e+00,
2.16294914e+00,
-1.48101497e-01};
EXPECT(migraphx::verify::verify_range(results_vector, gold));
EXPECT(migraphx::verify::verify_range_with_threshold(
results_vector, migraphx::verify::expected{gold}, migraphx::verify::threshold{9e-6}));
}
TEST_CASE_REGISTER(dot_2d_test<float>)
TEST_CASE_REGISTER(dot_2d_test<double>)
......@@ -93,7 +94,7 @@ void dot_4d_test()
1.33628943, -0.11413, -0.31270559, 1.59336732, -0.19361027,
-0.91620867, 0.40108416, -0.06969921, 0.68483471, -0.39906632,
-1.66423624, 0.69040076, -1.31490171, -0.11282616, -0.79391814};
std::vector<float> b = {6.09568541e-01,
std::vector<T> b = {6.09568541e-01,
-6.10527007e-01,
3.66646462e-01,
1.18951101e-01,
......@@ -108,7 +109,7 @@ void dot_4d_test()
1.53027987e+00,
-3.81407415e-04,
-3.29650255e-01};
std::vector<float> gold = {-1.56327541e+00,
std::vector<T> gold = {-1.56327541e+00,
-7.09570140e-01,
-5.37424982e-01,
-2.22994831e-01,
......@@ -129,8 +130,10 @@ void dot_4d_test()
auto result = p.eval({}).back();
std::vector<T> results_vector;
result.visit([&](auto output) { results_vector.assign(output.begin(), output.end()); });
EXPECT(migraphx::verify::verify_range(results_vector, gold));
EXPECT(migraphx::verify::verify_range_with_threshold(
results_vector, migraphx::verify::expected{gold}, migraphx::verify::threshold{9e-6}));
}
TEST_CASE_REGISTER(dot_4d_test<float>)
TEST_CASE_REGISTER(dot_4d_test<double>)
......
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