DragonX/dragonx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Dynamically determine thread count for dataset init

Browse Source
This commit is contained in:
Duke
2024-01-06 22:54:14 -08:00
parent 56ef56b136
commit 1b5422eb2d
1 changed files with 22 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
src/miner.cpp
View File

@@ -1201,14 +1201,27 @@ void static RandomXMiner()
rxdebug("%s: initialized cache with keyHeight=%d, randomxBlockKey=%s\n", keyHeight, randomxBlockKey.ToString().c_str()); rxdebug("%s: initialized cache with keyHeight=%d, randomxBlockKey=%s\n", keyHeight, randomxBlockKey.ToString().c_str());
} }
//TODO: this is hardcoded to use 2 threads instead of the number of mining threads int initThreadCount = std::thread::hardware_concurrency();
rxdebug("%s: initializing dataset with 2 threads\n"); if(initThreadCount > 1) {
std::thread t1(&randomx_init_dataset, randomxDataset, randomxCache, 0, datasetItemCount / 2); rxdebug("%s: initializing dataset with %d threads\n", initThreadCount);
std::thread t2(&randomx_init_dataset, randomxDataset, randomxCache, datasetItemCount / 2, datasetItemCount - datasetItemCount / 2); std::vector<std::thread> threads;
t1.join(); uint32_t startItem = 0;
t2.join(); auto perThread = datasetItemCount / initThreadCount;
auto remainder = datasetItemCount % initThreadCount;
for (int i = 0; i < initThreadCount; ++i) {
auto count = perThread + (i == initThreadCount - 1 ? remainder : 0);
threads.push_back(std::thread(&randomx_init_dataset, randomxDataset, randomxCache, startItem, count));
startItem += count;
}
for (unsigned i = 0; i < threads.size(); ++i) {
threads[i].join();
}
threads.clear();
} else {
rxdebug("%s: initializing dataset with 1 thread\n");
randomx_init_dataset(randomxDataset, randomxCache, 0, datasetItemCount);
}
// randomx_init_dataset(randomxDataset, randomxCache, 0, datasetItemCount);
rxdebug("%s: dataset initialized\n"); rxdebug("%s: dataset initialized\n");
myVM = randomx_create_vm(flags, nullptr, randomxDataset); myVM = randomx_create_vm(flags, nullptr, randomxDataset);
@@ -1463,8 +1476,8 @@ void static RandomXMiner()
randomx_release_dataset(randomxDataset); randomx_release_dataset(randomxDataset);
rxdebug("%s: released dataset in normal exit\n"); rxdebug("%s: released dataset in normal exit\n");
randomx_release_cache(randomxCache); //randomx_release_cache(randomxCache);
rxdebug("%s: released cache in normal exit\n"); //rxdebug("%s: released cache in normal exit\n");
miningTimer.stop(); miningTimer.stop();
c.disconnect(); c.disconnect();
} }