提交 a68255f9 编写于 作者: F fishpenguin

clang format


Former-commit-id: 3f99d7df6b042683696dd70c9dc0f14fee57b388
上级 68d5693a
......@@ -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;
}
......
......@@ -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>;
......
......@@ -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;
}
......
......@@ -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;
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册