Commit 4431d8f7 authored by peng.xu's avatar peng.xu
Browse files

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

fix scheduler unittest error

See merge request megasearch/milvus!563

Former-commit-id: c7e536745ffb08f12780ddac1cec2ae1177cafea
parents 369860eb aa5c0ba7

cpp/unittest/scheduler/scheduler_test.cpp

+4 −0
Original line number Diff line number Diff line
@@ -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);