diff --git a/src/tim/vx/ops/elementwise_test.cc b/src/tim/vx/ops/elementwise_test.cc index 02bc543..d4940fd 100644 --- a/src/tim/vx/ops/elementwise_test.cc +++ b/src/tim/vx/ops/elementwise_test.cc @@ -315,7 +315,7 @@ TEST(Div, Div_int32) { } std::vector output_data; - std::vector golden = {0, -1, 5, 1}; + std::vector golden = {0, -1, 5, 2}; output_data.resize(golden.size()); if (!output->CopyDataFromTensor(output_data.data())) { std::cout << "Copy output data fail." << std::endl; @@ -361,7 +361,7 @@ TEST(Div, Div_int32_broadcast) { } std::vector output_data; - std::vector golden = {-6, 7, 2, 2, 3, -41, -14, -16}; + std::vector golden = {-7, 7, 2, 3, 4, -41, -14, -16}; output_data.resize(golden.size()); if (!output->CopyDataFromTensor(output_data.data())) { std::cout << "Copy output data fail." << std::endl;