提交 ef267dac 编写于 作者: M Megvii Engine Team 提交者: Xu Xinran

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

GitOrigin-RevId: 5c557f082e70530bdf7187df324e8ff3edb8d2d5
上级 82692159
...@@ -70,9 +70,10 @@ bool eq_layout1(const TensorLayout &a, const TensorLayout &b) { ...@@ -70,9 +70,10 @@ bool eq_layout1(const TensorLayout &a, const TensorLayout &b) {
} }
} // anonymous namespace } // 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) { TEST(BENCHMARK_BASIC_TYPES, EQ_SHAPE) {
std::mt19937_64 rng; std::mt19937_64 rng;
static constexpr size_t NR_TEST = 1000000;
static TensorShape s0, s1[NR_TEST]; static TensorShape s0, s1[NR_TEST];
auto gen = [&](int type) { auto gen = [&](int type) {
if (type == 0) { if (type == 0) {
...@@ -115,7 +116,6 @@ TEST(BENCHMARK_BASIC_TYPES, EQ_SHAPE) { ...@@ -115,7 +116,6 @@ TEST(BENCHMARK_BASIC_TYPES, EQ_SHAPE) {
TEST(BENCHMARK_BASIC_TYPES, EQ_LAYOUT) { TEST(BENCHMARK_BASIC_TYPES, EQ_LAYOUT) {
std::mt19937_64 rng; std::mt19937_64 rng;
static constexpr size_t NR_TEST = 1000000;
static TensorLayout s0, s1[NR_TEST]; static TensorLayout s0, s1[NR_TEST];
auto gen = [&](int type) { auto gen = [&](int type) {
if (type == 0) { if (type == 0) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册