tqcq/trrespass
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Compare commits

base: tqcq:81f156be08a5625441ff758b196ab5dcb96a8d7e
Branches Tags
tqcq:master
..
compare: tqcq:a53cedd6d50bb99817c8c2fdf9ac02786924cdae
Branches Tags
tqcq:master

3 Commits
81f156be08 ... a53cedd6d5

Author SHA1 Message Date
tqcq
a53cedd6d5 feat: add log 2024-10-26 10:10:59 +00:00
tqcq
9b95b11f4d feat: update 2024-10-26 10:09:18 +00:00
tqcq
6374c128af fix: auto adjust thread num 2024-10-26 10:07:13 +00:00
1 changed files with 15 additions and 6 deletions
Show Stats Expand all files Collapse all files

21
drama/src/rev-mc.c
View File

@ -329,7 +329,7 @@ find_row_mask(std::vector<set_t> &sets,
std::vector<std::thread> workers;
int thread_num = cpu == 0 ? std::thread::hardware_concurrency() : cpu;
if (thread_num > 4) { thread_num = 4; }
if (cpu == 0 && thread_num > 4) { thread_num -= 4; }
fprintf(stderr, "thread_num: %d\n", thread_num);
uint64_t step = 1000;
@ -341,6 +341,7 @@ find_row_mask(std::vector<set_t> &sets,
// if (!step) { ++step; }
fprintf(stderr, "worker_step: %ld\n", step);
std::atomic<uint64_t> cnt{0};
for (int i = 0; i < thread_num; ++i) {
workers.emplace_back([&] {
while (!found) {
@ -357,16 +358,24 @@ find_row_mask(std::vector<set_t> &sets,
while (cur_mask < last_mask && step_count > 0 && !found.load(std::memory_order_relaxed)) {
for (int i = std::min(step_count, step); i > 0 && cur_mask < last_mask; --i) {
cur_mask = next_bit_permutation(cur_mask);
if (cnt.fetch_add(1) % 10000 == 0) {
fprintf(stderr, "cnt : %5ld\n", cnt.load(std::memory_order_relaxed));
}
}
step_count -= std::min(step_count, step);
std::lock_guard<std::mutex> _(lock);
if (my_pos > g_base_pos) {
g_base_pos = my_pos;
row_mask = cur_mask;
std::lock_guard<std::mutex> _(lock);
if (my_pos > g_base_pos) {
g_base_pos = my_pos;
row_mask = cur_mask;
}
} else if (g_base_pos <= g_cur_pos) {
cur_mask = row_mask;
my_pos = g_base_pos;
std::lock_guard<std::mutex> _(lock);
if (g_base_pos < g_cur_pos) {
cur_mask = row_mask;
my_pos = g_base_pos;
}
}
}