diff --git a/cpp/src/db/FaissSerializer.cpp b/cpp/src/db/FaissSerializer.cpp index fee77504692afdc27f332583dc95cf06dbf3140f..b2abb3a19c14936412f4c0bdc2f96af29ff5b5df 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 a56779996e8f6c3adc22c292669321862bf85043..d3a255a0bb91548e7ecad80d071e2cb82fb609a1 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 cb2891be2e65f6b7f8dfe720c732569195ff6e8d..dcb32e0b0538f67c671bbc1fc657b1f563c2928b 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() {} };