From 06d4747b312babf942d74027087505644f0304e4 Mon Sep 17 00:00:00 2001 From: Feiyue Chen Date: Wed, 28 Dec 2022 15:12:54 +0800 Subject: [PATCH] Fixed wrong type of hashtable_lookup unit test Modified datatype of copied output so that actual float16 output can be checked correctly Type: Bug Fix Issue: Bug 34696 Signed-off-by: Feiyue Chen --- src/tim/vx/ops/hashtable_lookup_test.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/tim/vx/ops/hashtable_lookup_test.cc b/src/tim/vx/ops/hashtable_lookup_test.cc index e743da9..04797e7 100644 --- a/src/tim/vx/ops/hashtable_lookup_test.cc +++ b/src/tim/vx/ops/hashtable_lookup_test.cc @@ -61,13 +61,13 @@ TEST(HashtableLookup, int32) { std::vector value_data = { 7,8,9,10 }; - std::vector out_golden = {7,0}; + std::vector out_golden = {18176,0};//check for float16 output std::vector hit_golden = {1,0};//0 means not hit EXPECT_TRUE(in_tensor->CopyDataToTensor(in_data.data(), in_data.size() * sizeof(int32_t))); EXPECT_TRUE(key_tensor->CopyDataToTensor(key_data.data(), key_data.size() * sizeof(int32_t))); EXPECT_TRUE(value_tensor->CopyDataToTensor(value_data.data(), value_data.size() * sizeof(int32_t))); - EXPECT_TRUE(out_tensor->CopyDataToTensor(out_golden.data(), out_golden.size() * sizeof(float))); + EXPECT_TRUE(out_tensor->CopyDataToTensor(out_golden.data(), out_golden.size())); EXPECT_TRUE(hit_tensor->CopyDataToTensor(hit_golden.data(), hit_golden.size() * sizeof(int32_t))); auto op = graph->CreateOperation(); @@ -76,7 +76,7 @@ TEST(HashtableLookup, int32) { EXPECT_TRUE(graph->Compile()); EXPECT_TRUE(graph->Run()); - std::vector output(out_golden.size()); + std::vector output(out_golden.size()); std::vector hit(hit_golden.size()); EXPECT_TRUE(out_tensor->CopyDataFromTensor(output.data())); EXPECT_TRUE(hit_tensor->CopyDataFromTensor(hit.data()));