Skip to content

Commit 1c3666a

Browse files
committed
Fixed benchmark code
1 parent b4dca14 commit 1c3666a

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

Diff for: src/tests/benchmark.cpp

+4-3
Original file line numberDiff line numberDiff line change
@@ -180,9 +180,10 @@ int main(int argc, char** argv) {
180180
std::cout << "ERROR: Dataset allocation failed" << std::endl;
181181
return 1;
182182
}
183+
uint32_t datasetItemCount = randomx_dataset_item_count();
183184
if (initThreadCount > 1) {
184-
auto perThread = RANDOMX_DATASET_ITEMS / initThreadCount;
185-
auto remainder = RANDOMX_DATASET_ITEMS % initThreadCount;
185+
auto perThread = datasetItemCount / initThreadCount;
186+
auto remainder = datasetItemCount % initThreadCount;
186187
uint32_t startItem = 0;
187188
for (int i = 0; i < initThreadCount; ++i) {
188189
auto count = perThread + (i == initThreadCount - 1 ? remainder : 0);
@@ -194,7 +195,7 @@ int main(int argc, char** argv) {
194195
}
195196
}
196197
else {
197-
randomx_init_dataset(dataset, cache, 0, RANDOMX_DATASET_ITEMS);
198+
randomx_init_dataset(dataset, cache, 0, datasetItemCount);
198199
}
199200
randomx_release_cache(cache);
200201
threads.clear();

0 commit comments

Comments
 (0)