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

feat optim

Browse Source
This commit is contained in:
tqcq 2024-10-24 06:44:00 +00:00
parent 4b572d8cfa
commit a7d47c26f6
1 changed files with 27 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -337,7 +337,7 @@ find_row_mask(std::vector<set_t> &sets,
fprintf(stderr, "worker_step: %ld\n", step); fprintf(stderr, "worker_step: %ld\n", step);
for (int i = 0; i < thread_num; ++i) { for (int i = 0; i < thread_num; ++i) {
workers.emplace_back([&] { workers.emplace_back([=, &row_mask, &found, &lock] {
while (!found) { while (!found) {
uint64_t cur_mask = last_mask; uint64_t cur_mask = last_mask;
{ {
@ -350,12 +350,33 @@ find_row_mask(std::vector<set_t> &sets,
} }
for (int i = 0; i < step && cur_mask < last_mask; ++i) { for (int i = 0; i < step && cur_mask < last_mask; ++i) {
if (!found.load(std::memory_order_relaxed) && resolve(cur_mask)) { if (found.load(std::memory_order_relaxed)) { break; }
std::lock_guard<std::mutex> _(lock);
if (!found.exchange(true)) { if (cur_mask & LS_BITMASK(CL_SHIFT)) {
row_mask = cur_mask; cur_mask = next_bit_permutation(row_mask);
break; continue;
}
bool mismatch = false;
for (auto addr_pool : same_row_sets) {
addr_tuple base_addr = addr_pool[0];
for (int i = 1; i < addr_pool.size(); i++) {
addr_tuple tmp = addr_pool[i];
if ((tmp.p_addr & row_mask) != (base_addr.p_addr & cur_mask)) {
mismatch = true;
break;
}
} }
if (mismatch) { break; }
}
if (mismatch) {
cur_mask = next_bit_permutation(row_mask);
continue;
}
std::lock_guard<std::mutex> _(lock);
if (!found.exchange(true)) {
row_mask = cur_mask;
break;
} }
} }
} }