diff --git a/cpp/src/db/FaissSerializer.cpp b/cpp/src/db/FaissSerializer.cpp index fee7750469..b2abb3a19c 100644 --- a/cpp/src/db/FaissSerializer.cpp +++ b/cpp/src/db/FaissSerializer.cpp @@ -1,6 +1,7 @@ #include #include #include +#include #include "FaissSerializer.h" @@ -33,6 +34,12 @@ Status FaissSerializer::Serialize() { return Status::OK(); } +Status FaissSerializer::Cache() { + zilliz::vecwise::cache::CpuCacheMgr::GetInstance( + )->InsertItem(location_, std::make_shared(pIndex_)); + + return Status::OK(); +} } // namespace engine } // namespace vecwise diff --git a/cpp/src/db/FaissSerializer.h b/cpp/src/db/FaissSerializer.h index a56779996e..d3a255a0bb 100644 --- a/cpp/src/db/FaissSerializer.h +++ b/cpp/src/db/FaissSerializer.h @@ -24,6 +24,8 @@ public: virtual Status Serialize() override; + virtual Status Cache() override; + protected: std::shared_ptr pIndex_; std::string location_; diff --git a/cpp/src/db/Serializer.h b/cpp/src/db/Serializer.h index cb2891be2e..dcb32e0b05 100644 --- a/cpp/src/db/Serializer.h +++ b/cpp/src/db/Serializer.h @@ -22,6 +22,8 @@ public: virtual Status Serialize() = 0; + virtual Status Cache() = 0; + virtual ~Serializer() {} };