From 7a16cb91c1cba69093d0fb9d4833e122d34cafe8 Mon Sep 17 00:00:00 2001 From: jinhai Date: Tue, 10 Sep 2019 12:10:14 +0800 Subject: [PATCH] Print log Former-commit-id: 5ccaed23bdb8446a7b4116ae46e0d7fb73fb2154 --- cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp b/cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp index bb89efbcd4..1474ff2e11 100644 --- a/cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp +++ b/cpp/src/core/src/knowhere/index/vector_index/gpu_ivf.cpp @@ -280,12 +280,15 @@ void FaissGpuResourceMgr::InitResource() { is_init = true; + std::cout << "InitResource" << std::endl; for(auto& device : devices_params_) { auto& device_id = device.first; + std::cout << "Device Id: " << device_id << std::endl; auto& device_param = device.second; auto& bq = idle_map[device_id]; for (int64_t i = 0; i < device_param.resource_num; ++i) { + std::cout << "Resource Id: " << i << std::endl; auto raw_resource = std::make_shared(); // TODO(linxj): enable set pinned memory @@ -295,6 +298,7 @@ void FaissGpuResourceMgr::InitResource() { bq.Put(res_wrapper); } } + std::cout << "End initResource" << std::endl; } ResPtr FaissGpuResourceMgr::GetRes(const int64_t &device_id,