mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
71ca600af9
Summary: Since caffe2 and torch have been consolidated, CAFFE2_API should be merged with TORCH_API. Addresses a TODO. Manually edited some references of the removed `CAFFE2_API`: * `CONTRIBUTING.md` * `caffe2/proto/CMakeLists.txt` * `cmake/ProtoBuf.cmake` * `c10/macros/Export.h` * `torch/csrc/WindowsTorchApiMacro.h` Pull Request resolved: https://github.com/pytorch/pytorch/pull/49496 Reviewed By: malfet, samestep Differential Revision: D25600726 Pulled By: janeyx99 fbshipit-source-id: 7e068d959e397ac183c097d7e9a9afeca5ddd782
40 lines
1.1 KiB
C++
40 lines
1.1 KiB
C++
#pragma once
|
|
|
|
#include <caffe2/distributed/store_handler.h>
|
|
|
|
namespace caffe2 {
|
|
|
|
class TORCH_API FileStoreHandler : public StoreHandler {
|
|
public:
|
|
explicit FileStoreHandler(const std::string& path, const std::string& prefix);
|
|
virtual ~FileStoreHandler();
|
|
|
|
virtual void set(const std::string& name, const std::string& data) override;
|
|
|
|
virtual std::string get(
|
|
const std::string& name,
|
|
const std::chrono::milliseconds& timeout = kDefaultTimeout) override;
|
|
|
|
virtual int64_t add(const std::string& name, int64_t value) override;
|
|
|
|
virtual bool deleteKey(const std::string& key) override;
|
|
|
|
virtual int64_t getNumKeys() override;
|
|
|
|
virtual bool check(const std::vector<std::string>& names) override;
|
|
|
|
virtual void wait(
|
|
const std::vector<std::string>& names,
|
|
const std::chrono::milliseconds& timeout = kDefaultTimeout) override;
|
|
|
|
protected:
|
|
std::string basePath_;
|
|
|
|
std::string realPath(const std::string& path);
|
|
|
|
std::string tmpPath(const std::string& name);
|
|
|
|
std::string objectPath(const std::string& name);
|
|
};
|
|
|
|
} // namespace caffe2
|