Browse Source

fix(test): fix test for brainpp docker env

GitOrigin-RevId: c4c2cc73d2
HuaHua404-patch-4
Megvii Engine Team 2 years ago
parent
commit
626222c698
2 changed files with 4 additions and 3 deletions
  1. +3
    -2
      src/core/test/comp_node.cpp
  2. +1
    -1
      src/core/test/utils/async_worker.cpp

+ 3
- 2
src/core/test/comp_node.cpp View File

@@ -563,10 +563,11 @@ TEST(TestCompNodeSyncManager, DeviceWait) {
cn2.device_wait_event(mgr.busy_wait_set_ready_and_get_event());
ev_cn1->record();
ev_cn2_end->record();
EXPECT_LE(timer.get_secs(), 0.06);
std::this_thread::yield();
EXPECT_LE(timer.get_secs(), 0.09);

ev_cn1->host_wait();
EXPECT_GE(timer.get_secs(), 0.1);
EXPECT_GE(timer.get_secs(), 0.12);
ev_cn2_end->host_wait();
auto ev2_t = ev_cn2_begin->elapsed_time_until(*ev_cn2_end);
EXPECT_GE(ev2_t, 0.1);


+ 1
- 1
src/core/test/utils/async_worker.cpp View File

@@ -87,8 +87,8 @@ TEST(TestAsyncWorker, AsyncWorkerSetException) {
std::this_thread::sleep_for(100ms);
});

timer.reset();
worker_set.start();
timer.reset();
ASSERT_THROW(worker_set.wait_all(), std::runtime_error);
ASSERT_LT(timer.get_msecs(), 100);
}


Loading…
Cancel
Save