#1946 fix load index file CPU2GPU fail during searching (#2182)

Signed-off-by: Nfeisiyicl <7764126@qq.com>

CHANGELOG.md

@@ -7,6 +7,7 @@ Please mark all change in change log and use the issue from GitHub
 ## Bug
 -   \#1705 Limit the insert data batch size
 -   \#1929 Skip MySQL meta schema field width check
+-   \#1946 Fix load index file CPU2GPU fail during searching
 -   \#1997 Index file missed after compact
 -   \#2073 Fix CheckDBConfigBackendUrl error message
 -   \#2076 CheckMetricConfigAddress error message

core/src/index/knowhere/knowhere/index/vector_index/helpers/FaissGpuResourceMgr.cpp

@@ -51,11 +51,10 @@ FaissGpuResourceMgr::InitDevice(int64_t device_id, int64_t pin_mem_size, int64_t
 
 void
 FaissGpuResourceMgr::InitResource() {
-    if (is_init)
-        return;
-
-    is_init = true;
+    if (!initialized_) {
+        std::lock_guard<std::mutex> lock(init_mutex_);
 
+        if (!initialized_) {
             for (auto& device : devices_params_) {
                 auto& device_id = device.first;
 
@@ -75,6 +74,9 @@ FaissGpuResourceMgr::InitResource() {
                 }
                 LOG_KNOWHERE_DEBUG_ << "DEVICEID " << device_id << ", resource count " << bq.Size();
             }
+            initialized_ = true;
+        }
+    }
 }
 
 ResPtr
@@ -115,7 +117,7 @@ FaissGpuResourceMgr::Free() {
             bq.Take();
         }
     }
-    is_init = false;
+    initialized_ = false;
 }
 
 void

core/src/index/knowhere/knowhere/index/vector_index/helpers/FaissGpuResourceMgr.h

@@ -77,7 +77,8 @@ class FaissGpuResourceMgr {
     Dump();
 
  protected:
-    bool is_init = false;
+    bool initialized_ = false;
+    std::mutex init_mutex_;
 
     std::map<int64_t, std::unique_ptr<std::mutex>> mutex_cache_;
     std::map<int64_t, DeviceParams> devices_params_;