diff --git a/dnn/test/common/benchmark_basic_types.cpp b/dnn/test/common/benchmark_basic_types.cpp index d467cf599015fc777bc048049a1f50d75381b944..13c98ac25a657a6c4dabe40829b48d460a24c76c 100644 --- a/dnn/test/common/benchmark_basic_types.cpp +++ b/dnn/test/common/benchmark_basic_types.cpp @@ -70,9 +70,10 @@ bool eq_layout1(const TensorLayout &a, const TensorLayout &b) { } } // anonymous namespace +// config NR_TEST at small memory device, eg, EV300 etc +static constexpr size_t NR_TEST = 10000; TEST(BENCHMARK_BASIC_TYPES, EQ_SHAPE) { std::mt19937_64 rng; - static constexpr size_t NR_TEST = 1000000; static TensorShape s0, s1[NR_TEST]; auto gen = [&](int type) { if (type == 0) { @@ -115,7 +116,6 @@ TEST(BENCHMARK_BASIC_TYPES, EQ_SHAPE) { TEST(BENCHMARK_BASIC_TYPES, EQ_LAYOUT) { std::mt19937_64 rng; - static constexpr size_t NR_TEST = 1000000; static TensorLayout s0, s1[NR_TEST]; auto gen = [&](int type) { if (type == 0) {