mirror of https://github.com/milvus-io/milvus.git
parent
49a5cdf682
commit
384b09e993
|
@ -180,6 +180,8 @@ TaskTable::PickToLoad(uint64_t limit) {
|
|||
cross = true;
|
||||
indexes.push_back(index);
|
||||
++pick_count;
|
||||
} else {
|
||||
cross = true;
|
||||
}
|
||||
}
|
||||
// rc.ElapseFromBegin("PickToLoad ");
|
||||
|
@ -252,6 +254,8 @@ TaskTable::PickToExecute(uint64_t limit) {
|
|||
cross = true;
|
||||
indexes.push_back(index);
|
||||
++pick_count;
|
||||
} else {
|
||||
cross = true;
|
||||
}
|
||||
}
|
||||
// rc.ElapseFromBegin("PickToExecute ");
|
||||
|
|
|
@ -172,10 +172,14 @@ Resource::loader_function() {
|
|||
}
|
||||
LoadFile(task_item->task);
|
||||
task_item->Loaded();
|
||||
if (task_item->from) {
|
||||
task_item->from->Moved();
|
||||
task_item->from->task = FinishedTask::Create(task_item->from->task);
|
||||
task_item->from = nullptr;
|
||||
|
||||
auto& label = task_item->task->label();
|
||||
if (label != nullptr && label->Type() != TaskLabelType::BROADCAST) {
|
||||
if (task_item->from) {
|
||||
task_item->from->Moved();
|
||||
task_item->from->task = FinishedTask::Create(task_item->from->task);
|
||||
task_item->from = nullptr;
|
||||
}
|
||||
}
|
||||
if (subscriber_) {
|
||||
auto event = std::make_shared<LoadCompletedEvent>(shared_from_this(), task_item);
|
||||
|
|
|
@ -222,8 +222,6 @@ if (MILVUS_GPU_VERSION)
|
|||
link_directories("${CUDA_TOOLKIT_ROOT_DIR}/lib64")
|
||||
set(unittest_libs ${unittest_libs}
|
||||
${CUDA_TOOLKIT_ROOT_DIR}/lib64/stubs/libnvidia-ml.so
|
||||
cudart
|
||||
cublas
|
||||
)
|
||||
endif ()
|
||||
|
||||
|
|
Loading…
Reference in New Issue