diff options
Diffstat (limited to 'include/util/types.h')
| -rw-r--r-- | include/util/types.h | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/include/util/types.h b/include/util/types.h index e67c486..72b3d29 100644 --- a/include/util/types.h +++ b/include/util/types.h @@ -111,7 +111,11 @@ public: void add_reconstruction(level_index source, level_index target, size_t reccnt, ReconstructionType type) { - m_tasks.push_back({{{source, all_shards_idx}}, target, reccnt}); + if (type == ReconstructionType::Merge) { + m_tasks.push_back({{{source, all_shards_idx}, {target, all_shards_idx}}, target, reccnt}); + } else { + m_tasks.push_back({{{source, all_shards_idx}}, target, reccnt}); + } total_reccnt += reccnt; } |