#4174 Search out of memory: CPU2GPU1 with index flat (#4178)

* Fix gpu resource issue

Signed-off-by: sahuang <xiaohai.xu@zilliz.com>

* fix_ut

Signed-off-by: shengjun.li <shengjun.li@zilliz.com>

Co-authored-by: shengjun.li <shengjun.li@zilliz.com>

CHANGELOG.md

@@ -9,6 +9,7 @@ Please mark all change in change log and use the issue from GitHub
 -   \#3906 Change DeleteTask state when it is loaded to avoid server crash.
 -   \#4012 Milvus hangs when continually creating and dropping partitions.
 -   \#4075 Improve performance for create large amount of partitions
+-   \#4174 Search out of memory: CPU2GPU1 with index flat
 
 ## Feature
 -   \#3773 Support IVF_PQ to run on FPGA.

core/src/scheduler/TaskTable.cpp

@@ -165,7 +165,7 @@ TaskTable::PickToLoad(uint64_t limit) {
         } else if (table_[index]->state == TaskTableItemState::LOADED) {
             cross = true;
             ++loaded_count;
-            if (loaded_count > 2)
+            if (loaded_count >= 1)
                 return std::vector<uint64_t>();
         } else if (table_[index]->state == TaskTableItemState::START) {
             auto task = table_[index]->task;

core/unittest/scheduler/test_resource.cpp

@@ -24,7 +24,7 @@
 namespace milvus {
 namespace scheduler {
 
-constexpr uint64_t max_once_load = 2;
+constexpr uint64_t max_once_load = 1;
 
 /************ ResourceBaseTest ************/
 class ResourceBaseTest : public testing::Test {