mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-01 16:35:29 +08:00
86 lines
2.0 KiB
C++
86 lines
2.0 KiB
C++
/*******************************************************************************
|
|
* Copyright 上海赜睿信息科技有限公司(Zilliz) - All Rights Reserved
|
|
* Unauthorized copying of this file, via any medium is strictly prohibited.
|
|
* Proprietary and confidential.
|
|
******************************************************************************/
|
|
#pragma once
|
|
|
|
#include "db/Status.h"
|
|
|
|
#include <vector>
|
|
#include <memory>
|
|
|
|
namespace zilliz {
|
|
namespace milvus {
|
|
namespace engine {
|
|
|
|
enum class EngineType {
|
|
INVALID = 0,
|
|
FAISS_IDMAP = 1,
|
|
FAISS_IVFFLAT,
|
|
FAISS_IVFSQ8,
|
|
NSG_MIX,
|
|
MAX_VALUE = NSG_MIX,
|
|
};
|
|
|
|
enum class MetricType {
|
|
L2 = 1,
|
|
IP = 2,
|
|
};
|
|
|
|
class ExecutionEngine {
|
|
public:
|
|
|
|
virtual Status AddWithIdArray(const std::vector<float>& vectors, const std::vector<long>& vector_ids);
|
|
|
|
virtual Status AddWithIds(long n, const float *xdata, const long *xids) = 0;
|
|
|
|
virtual size_t Count() const = 0;
|
|
|
|
virtual size_t Size() const = 0;
|
|
|
|
virtual size_t Dimension() const = 0;
|
|
|
|
virtual size_t PhysicalSize() const = 0;
|
|
|
|
virtual Status Serialize() = 0;
|
|
|
|
virtual Status Load(bool to_cache = true) = 0;
|
|
|
|
virtual Status CopyToGpu(uint64_t device_id) = 0;
|
|
|
|
virtual Status CopyToCpu() = 0;
|
|
|
|
virtual std::shared_ptr<ExecutionEngine> Clone() = 0;
|
|
|
|
virtual Status Merge(const std::string& location) = 0;
|
|
|
|
virtual Status Search(long n,
|
|
const float *data,
|
|
long k,
|
|
long nprobe,
|
|
float *distances,
|
|
long *labels) const = 0;
|
|
|
|
virtual std::shared_ptr<ExecutionEngine> BuildIndex(const std::string& location, EngineType engine_type) = 0;
|
|
|
|
virtual Status Cache() = 0;
|
|
|
|
virtual Status GpuCache(uint64_t gpu_id) = 0;
|
|
|
|
virtual Status Init() = 0;
|
|
|
|
virtual EngineType IndexEngineType() const = 0;
|
|
|
|
virtual MetricType IndexMetricType() const = 0;
|
|
|
|
virtual std::string GetLocation() const = 0;
|
|
};
|
|
|
|
using ExecutionEnginePtr = std::shared_ptr<ExecutionEngine>;
|
|
|
|
|
|
} // namespace engine
|
|
} // namespace milvus
|
|
} // namespace zilliz
|