admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2026-01-05 18:31:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'branch-0.4.0' into 'branch-0.4.0'

Browse Source 
fix scheduler unittest error

See merge request megasearch/milvus!563

Former-commit-id: e5048bd9d4685ca47f403b19acdd118a4442791b
This commit is contained in:
peng.xu 2019-09-12 17:10:11 +08:00
commit 6d6bd52c24
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
cpp/unittest/scheduler/scheduler_test.cpp
View File

@ -94,6 +94,10 @@ class SchedulerTest : public testing::Test {
protected:
void
SetUp() override {
constexpr int64_t cache_cap = 1024*1024*1024;
cache::GpuCacheMgr::GetInstance(0)->SetCapacity(cache_cap);
cache::GpuCacheMgr::GetInstance(1)->SetCapacity(cache_cap);
ResourcePtr cpu = ResourceFactory::Create("cpu", "CPU", 0, true, false);
ResourcePtr gpu_0 = ResourceFactory::Create("gpu0", "GPU", 0);
ResourcePtr gpu_1 = ResourceFactory::Create("gpu1", "GPU", 1);