Browse Source

fix(megdnn_test/ev300): try run megdnn_test on ev300 board

GitOrigin-RevId: 5c557f082e
tags/v1.0.0-rc1
Megvii Engine Team Xu Xinran 4 years ago
parent
commit
ef267dacf8
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      dnn/test/common/benchmark_basic_types.cpp

+ 2
- 2
dnn/test/common/benchmark_basic_types.cpp View File

@@ -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) {


Loading…
Cancel
Save