fix stucked tasiks (#3666)

Signed-off-by: shengjun.li <shengjun.li@zilliz.com>
pull/3677/head^2
shengjun.li 2020-09-10 09:14:03 +08:00 committed by GitHub
parent 49a5cdf682
commit 384b09e993
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 6 deletions

View File

@ -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 ");

View File

@ -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);

View File

@ -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 ()