Prevent an out-of-bounds load in connectivity sets

kahypar · Sep 22, 2024 · 70424b0 · 70424b0
1 parent afcf4b4
commit 70424b0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mt-kahypar/datastructures/connectivity_set.h b/mt-kahypar/datastructures/connectivity_set.h
@@ -87,7 +87,9 @@ class ConnectivitySets {
     _shallow_copy_bitset() {
       if ( num_hyperedges > 0 ) {
         _bits.resize("Refinement", "connectivity_set",
-          static_cast<size_t>(num_hyperedges) * _num_blocks_per_hyperedge, true, assign_parallel);
+          static_cast<size_t>(num_hyperedges) * _num_blocks_per_hyperedge
+          + 1 // The nextBlockID() implementation performs a (masked out) load past the end
+          , true, assign_parallel);
       }
     }