milvus/cpp/src/scheduler/resource/TestResource.h
wxyu 18dd06d07a MS-511 Update resource_test in scheduler
Former-commit-id: 99e298907ae56cfeb2475431080d06c4b34216f7
2019-09-07 16:45:17 +08:00

39 lines
919 B
C++

/*******************************************************************************
* Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
* Unauthorized copying of this file, via any medium is strictly prohibited.
* Proprietary and confidential.
******************************************************************************/
#pragma once
#include "Resource.h"
namespace zilliz {
namespace milvus {
namespace engine {
class TestResource : public Resource {
public:
explicit
TestResource(std::string name, uint64_t device_id, bool enable_loader, bool enable_executor);
inline std::string
Dump() const override {
return "<TestResource, name=" + name_ + ">";
}
friend std::ostream &operator<<(std::ostream &out, const TestResource &resource);
protected:
void
LoadFile(TaskPtr task) override;
void
Process(TaskPtr task) override;
};
}
}
}