Skip to content

Commit 7b2c064

Browse files
BobLiu20rhdong
authored andcommitted
fix redis cluster MaccumCommand
1 parent f97108e commit 7b2c064

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

tensorflow_recommenders_addons/dynamic_embedding/core/kernels/redis_impl/redis_cluster_connection_pool.hpp

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1302,14 +1302,15 @@ every bucket has its own BucketContext for sending data---for locating reply-
13021302
exists_split[key_bucket_locs].push_back(*(exists + begin + i));
13031303
}
13041304

1305-
const bool *pe_raw = exists + begin;
1305+
std::vector<std::unique_ptr<std::vector<char>>> exists_chars(storage_slice);
13061306
for (unsigned i = 0; i < storage_slice; ++i) {
13071307
if (!exists_split[i].empty()) {
1308-
std::vector<char> exists_char(exists_split[i].size());
1309-
std::transform(exists_split[i].begin(), exists_split[i].end(), exists_char.begin(), [](bool b) {
1308+
exists_chars[i] = std::make_unique<std::vector<char>>(exists_split[i].size());
1309+
std::transform(exists_split[i].begin(), exists_split[i].end(), exists_chars[i]->begin(), [](bool b) {
13101310
return static_cast<char>(b);
13111311
});
1312-
thread_context->HandlePushBack(i, exists_char.data(), exists_char.size() * sizeof(char));
1312+
1313+
thread_context->HandlePushBack(i, exists_chars[i]->data(), exists_chars[i]->size() * sizeof(char));
13131314
}
13141315
}
13151316

0 commit comments

Comments
 (0)