1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
|
#pragma once
#include <torch/csrc/distributed/c10d/Store.hpp>
#include "TestUtils.hpp"
#include <gtest/gtest.h>
namespace c10d {
namespace test {
inline void set(
Store& store,
const std::string& key,
const std::string& value) {
std::vector<uint8_t> data(value.begin(), value.end());
store.set(key, data);
}
inline std::vector<uint8_t> compareSet(
Store& store,
const std::string& key,
const std::string& expectedValue,
const std::string& desiredValue) {
std::vector<uint8_t> expectedData(expectedValue.begin(), expectedValue.end());
std::vector<uint8_t> desiredData(desiredValue.begin(), desiredValue.end());
return store.compareSet(key, expectedData, desiredData);
}
inline void check(
Store& store,
const std::string& key,
const std::string& expected) {
auto tmp = store.get(key);
auto actual = std::string((const char*)tmp.data(), tmp.size());
EXPECT_EQ(actual, expected);
}
inline void deleteKey(
Store& store,
const std::string& key) {
store.deleteKey(key);
}
} // namespace test
} // namespace c10d
|
