Skip to content

Commit 56ceea4

Browse files
Quick fixes: accuracy mode & C API
1 parent 2664da2 commit 56ceea4

File tree

3 files changed

+5
-4
lines changed

3 files changed

+5
-4
lines changed

loadgen/bindings/c_api.cc

+3-2
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ limitations under the License.
1313
#include "c_api.h"
1414

1515
#include <string>
16+
#include <cassert>
1617

1718
#include "../loadgen.h"
1819
#include "../query_sample.h"
@@ -148,7 +149,7 @@ class GroupedQuerySampleLibraryTrampoline : public QuerySampleLibrary {
148149

149150
total_sample_count_ = 0;
150151

151-
for(ssize_t i = 0; i < group_sizes.size(); i++){
152+
for(size_t i = 0; i < group_sizes.size(); i++){
152153
group_sizes_.push_back(group_sizes[i]);
153154
total_sample_count_ += group_sizes[i];
154155
for(size_t j = 0; j < group_sizes[i]; j++){
@@ -193,7 +194,7 @@ void* ConstructGroupedQSL(ClientData client_data, const char* name, size_t name_
193194
UnloadSamplesFromRamCallback unload_samples_from_ram_cb,
194195
std::vector<size_t>& group_sizes) {
195196
GroupedQuerySampleLibraryTrampoline* qsl = new GroupedQuerySampleLibraryTrampoline(
196-
client_data, std::string(name, name_length), total_sample_count,
197+
client_data, std::string(name, name_length),
197198
performance_sample_count, load_samples_to_ram_cb,
198199
unload_samples_from_ram_cb, group_sizes);
199200
return reinterpret_cast<void*>(qsl);

loadgen/bindings/python_api.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -155,7 +155,7 @@ class GroupedQuerySampleLibraryTrampoline : public QuerySampleLibrary {
155155
auto buffer = group_sizes.request();
156156
size_t* ptr = (size_t*)buffer.ptr;
157157

158-
for(ssize_t i = 0; i < group_sizes.shape()[0]; i++){
158+
for(size_t i = 0; i < group_sizes.shape()[0]; i++){
159159
group_sizes_.push_back(ptr[i]);
160160
total_sample_count_ += ptr[i];
161161
for(size_t j = 0; j < ptr[i]; j++){

loadgen/loadgen.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -751,7 +751,7 @@ std::vector<LoadableSampleSet> GenerateLoadableSets(
751751
size_t number_of_groups = qsl->NumberOfGroups();
752752
for (size_t i = 0; i < number_of_groups; i++) {
753753
size_t group_size = qsl->GroupSize(groupIdx[idx]);
754-
if (loadable_set.size() + group_size < set_size) {
754+
if (loadable_set.size() + group_size <= set_size) {
755755
for (size_t j = 0; j < group_size; j++) {
756756
loadable_set.push_back(samples[idx]);
757757
idx++;

0 commit comments

Comments
 (0)