diff options
Diffstat (limited to 'include/util')
| -rw-r--r-- | include/util/types.h | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/include/util/types.h b/include/util/types.h index e67c486..084bf4b 100644 --- a/include/util/types.h +++ b/include/util/types.h @@ -111,7 +111,13 @@ 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; } |