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
|
#include <c10/util/Synchronized.h>
#include <gtest/gtest.h>
#include <array>
#include <thread>
namespace {
TEST(Synchronized, TestSingleThreadExecution) {
c10::Synchronized<int> iv(0);
const int kMaxValue = 100;
for (int i = 0; i < kMaxValue; ++i) {
auto ret = iv.withLock([](int& iv) { return ++iv; });
EXPECT_EQ(ret, i + 1);
}
iv.withLock([kMaxValue](int& iv) { EXPECT_EQ(iv, kMaxValue); });
}
TEST(Synchronized, TestMultiThreadedExecution) {
c10::Synchronized<int> iv(0);
#define NUM_LOOP_INCREMENTS 10000
auto thread_cb = [&iv]() {
for (int i = 0; i < NUM_LOOP_INCREMENTS; ++i) {
iv.withLock([](int& iv) { ++iv; });
}
};
std::array<std::thread, 10> threads;
for (auto& t : threads) {
t = std::thread(thread_cb);
}
for (auto& t : threads) {
t.join();
}
iv.withLock([](int& iv) { EXPECT_EQ(iv, NUM_LOOP_INCREMENTS * 10); });
#undef NUM_LOOP_INCREMENTS
}
} // namespace
|
