提交 4431d8f7 编写于 作者: P peng.xu

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
......@@ -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);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册