diff --git a/lite/tests/kernels/reduce_max_compute_test.cc b/lite/tests/kernels/reduce_max_compute_test.cc index bf9d4ea720dcc73bdf4f01b6cad10644a72de9b1..769efcd156a6afdc4aedbbad2587cafd324e40cd 100644 --- a/lite/tests/kernels/reduce_max_compute_test.cc +++ b/lite/tests/kernels/reduce_max_compute_test.cc @@ -423,19 +423,13 @@ void test_reduce_max(Place place) { void test_reduce_max_for_three(Place place) { std::vector> reduce_dim{{0}, {1}, {2}}; - for (auto f : {1, 3}) { - for (auto s : {1, 2}) { - for (auto t : {1, 3}) { - for (bool keep_dim : {false, true}) { - for (auto dim : reduce_dim) { - auto x_dims = DDim(std::vector({f, s, t})); - std::unique_ptr tester(new ReduceMaxComputeTester( - place, "def", dim, keep_dim, x_dims)); - arena::Arena arena(std::move(tester), place, 2e-5); - arena.TestPrecision(); - } - } - } + for (bool keep_dim : {false, true}) { + for (auto dim : reduce_dim) { + auto x_dims = DDim(std::vector({3, 4, 5})); + std::unique_ptr tester( + new ReduceMaxComputeTester(place, "def", dim, keep_dim, x_dims)); + arena::Arena arena(std::move(tester), place, 2e-5); + arena.TestPrecision(); } } }