mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
d177846dbf
Summary: This brings it up to par with how the RedisStoreHandler works. The store handler configuration does not have to change and only the run ID parameter changes across runs. This was inconsistent and came up in https://github.com/caffe2/caffe2/issues/984. Reviewed By: Yangqing Differential Revision: D5539299 fbshipit-source-id: 3b5f31c6549b46c24bbd70ebc0bec150eac8b76c
34 lines
893 B
C++
34 lines
893 B
C++
#pragma once
|
|
|
|
#include <caffe2/distributed/store_handler.h>
|
|
|
|
namespace caffe2 {
|
|
|
|
class 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) override;
|
|
|
|
virtual int64_t add(const std::string& name, int64_t value) 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
|