Skip to content

GitLab

Commit a8232451 authored by Anthony Chang's avatar Anthony Chang
Browse files

post-rebase fix to accommodate check_err's new interface

parent 96f57e07
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
test/softmax/test_softmax_host_ref.cpp
View file @ a8232451
...@@ -28,7 +28,8 @@ TEST(ReferenceSoftmax, Run) ...@@ -28,7 +28,8 @@ TEST(ReferenceSoftmax, Run)
auto ref_softmax_argument = ref_softmax.MakeArgument(x, y, alpha, beta, {1}); auto ref_softmax_argument = ref_softmax.MakeArgument(x, y, alpha, beta, {1});
ref_softmax_invoker.Run(ref_softmax_argument); ref_softmax_invoker.Run(ref_softmax_argument);
EXPECT_TRUE((utils::check_err(y.mData, {0.73105858, 0.268941421, 0.26894142, 0.73105858}))); EXPECT_TRUE((utils::check_err(
y.mData, std::vector<float>{0.73105858f, 0.268941421f, 0.26894142f, 0.73105858f})));
} }
TEST(ReferenceSoftmax, RunWithCalculatedStats) TEST(ReferenceSoftmax, RunWithCalculatedStats)
...@@ -57,8 +58,8 @@ TEST(ReferenceSoftmax, RunWithCalculatedStats) ...@@ -57,8 +58,8 @@ TEST(ReferenceSoftmax, RunWithCalculatedStats)
{ {
auto ref_softmax_argument = ref_softmax.MakeArgument(x, y, alpha, beta, {1}, &stats); auto ref_softmax_argument = ref_softmax.MakeArgument(x, y, alpha, beta, {1}, &stats);
ref_softmax_invoker.Run(ref_softmax_argument); ref_softmax_invoker.Run(ref_softmax_argument);
EXPECT_TRUE( EXPECT_TRUE((utils::check_err(
(utils::check_err(stats.mData, {1.74366838, 1.74366838, 1.74366838, 1.74366838}))); stats.mData, std::vector<float>{1.74366838f, 1.74366838f, 1.74366838f, 1.74366838f})));
} }
{ {
......
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