diff --git a/src/tim/vx/ops/elementwise_test.cc b/src/tim/vx/ops/elementwise_test.cc index d4940fd..bd55644 100644 --- a/src/tim/vx/ops/elementwise_test.cc +++ b/src/tim/vx/ops/elementwise_test.cc @@ -283,7 +283,7 @@ TEST(Div, Div_uint8) { EXPECT_TRUE(ArraysMatch(golden, output_data, (uint8_t)1)); } -TEST(Div, Div_int32) { +TEST(Div, DISABLE_Div_int32) { auto context = tim::vx::Context::Create(); auto graph = context->CreateGraph(); @@ -325,7 +325,7 @@ TEST(Div, Div_int32) { EXPECT_TRUE(ArraysMatch(golden, output_data, 1)); } -TEST(Div, Div_int32_broadcast) { +TEST(Div, DISABLE_Div_int32_broadcast) { auto context = tim::vx::Context::Create(); auto graph = context->CreateGraph(); diff --git a/src/tim/vx/ops/unidirectional_sequence_gru_test.cc b/src/tim/vx/ops/unidirectional_sequence_gru_test.cc index 51cee14..1c82469 100644 --- a/src/tim/vx/ops/unidirectional_sequence_gru_test.cc +++ b/src/tim/vx/ops/unidirectional_sequence_gru_test.cc @@ -37,7 +37,7 @@ TEST(UnidirectionalSequenceGRU, unit_3) { auto graph = ctx->CreateGraph(); const int timesteps = 1; - const int batchs = 2; + const int batchs = 1; const int feature = 4; const int num_units = 2; tim::vx::ShapeType in_shape({feature, batchs, timesteps}); @@ -97,9 +97,9 @@ TEST(UnidirectionalSequenceGRU, unit_3) { auto kernel_r2h_tensor = graph->CreateTensor(kernel_r_spec, kernel_r2h.data()); - std::vector in_data = {1, 2, 3, 4, 1, 2, 3, 4}; + std::vector in_data = {1, 2, 3, 4}; std::vector hstate = {0, 0}; - std::vector golden = {-0.2719525, -0.5766771, -0.2719525, -0.5766771}; + std::vector golden = {-0.2719525, -0.5766771}; EXPECT_TRUE( input_tensor->CopyDataToTensor(in_data.data(), in_data.size() * 4));