admin/milvus
1
0
Fork 0
mirror of https://gitee.com/milvus-io/milvus.git synced 2025-12-30 15:35:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

clang format

Browse Source 
Former-commit-id: 3f99d7df6b042683696dd70c9dc0f14fee57b388
This commit is contained in:
fishpenguin 2019-10-31 14:44:59 +08:00
parent 68d5693a74
commit a68255f9d0
4 changed files with 16 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/src/scheduler/optimizer/OnlyCPUPass.cpp
View File

@ -17,9 +17,9 @@
#include "scheduler/optimizer/OnlyCPUPass.h"
#include "scheduler/SchedInst.h"
#include "scheduler/Utils.h"
#include "scheduler/task/SearchTask.h"
#include "scheduler/tasklabel/SpecResLabel.h"
#include "scheduler/Utils.h"
namespace milvus {
namespace scheduler {
@ -29,8 +29,8 @@ OnlyCPUPass::Run(const TaskPtr& task) {
if (task->Type() != TaskType::SearchTask)
return false;
auto search_task = std::static_pointer_cast<XSearchTask>(task);
if (search_task->file_->engine_type_ != (int) engine::EngineType::FAISS_IVFSQ8 &&
search_task->file_->engine_type_ != (int) engine::EngineType::FAISS_IVFFLAT) {
if (search_task->file_->engine_type_ != (int)engine::EngineType::FAISS_IVFSQ8 &&
search_task->file_->engine_type_ != (int)engine::EngineType::FAISS_IVFFLAT) {
return false;
}

8
core/src/scheduler/optimizer/OnlyCPUPass.h
View File

@ -32,13 +32,13 @@
namespace milvus {
namespace scheduler {
class OnlyCPUPass : public Pass {
public:
class OnlyCPUPass : public Pass {
public:
OnlyCPUPass() = default;
public:
public:
bool
Run(const TaskPtr &task) override;
Run(const TaskPtr& task) override;
};
using OnlyCPUPassPtr = std::shared_ptr<OnlyCPUPass>;

8
core/src/scheduler/optimizer/OnlyGPUPass.cpp
View File

@ -17,9 +17,9 @@
#include "scheduler/optimizer/OnlyGPUPass.h"
#include "scheduler/SchedInst.h"
#include "scheduler/Utils.h"
#include "scheduler/task/SearchTask.h"
#include "scheduler/tasklabel/SpecResLabel.h"
#include "scheduler/Utils.h"
#include "server/Config.h"
namespace milvus {
@ -31,15 +31,15 @@ OnlyGPUPass::Run(const TaskPtr& task) {
return false;
auto search_task = std::static_pointer_cast<XSearchTask>(task);
if (search_task->file_->engine_type_ != (int) engine::EngineType::FAISS_IVFSQ8 &&
search_task->file_->engine_type_ != (int) engine::EngineType::FAISS_IVFFLAT) {
if (search_task->file_->engine_type_ != (int)engine::EngineType::FAISS_IVFSQ8 &&
search_task->file_->engine_type_ != (int)engine::EngineType::FAISS_IVFFLAT) {
return false;
}
server::Config& config = server::Config::GetInstance();
std::vector<std::string> search_resources;
config.GetResourceConfigSearchResources(search_resources);
for (auto &resource : search_resources) {
for (auto& resource : search_resources) {
if (resource == "cpu") {
return false;
}

10
core/src/scheduler/optimizer/OnlyGPUPass.h
View File

@ -32,15 +32,15 @@
namespace milvus {
namespace scheduler {
class OnlyGPUPass : public Pass {
public:
class OnlyGPUPass : public Pass {
public:
OnlyGPUPass() = default;
public:
public:
bool
Run(const TaskPtr &task) override;
Run(const TaskPtr& task) override;
private:
private:
uint64_t specified_gpu_id_ = 0;
};