Bypassing a tile now preserves both distributed multicast stats

and (new) replication factor.
NVlabs · Sep 21, 2023 · ba2afa8 · ba2afa8
1 parent 36b791e
commit ba2afa8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/loop-analysis/tiling.cpp b/src/loop-analysis/tiling.cpp
@@ -294,8 +294,12 @@ void ProcessOuterMaskedLevels(std::vector<DataMovementInfo>& tile_nest, std::bit
       // network stats (because distributed network traffic is transferred
       // on behalf of the child level.
       auto saved_distributed_stats = tile_nest[cur].distributed_access_stats;
+      auto saved_replication_factor = tile_nest[cur].replication_factor;
+
       tile_nest[cur].Reset();
+
       tile_nest[cur].distributed_access_stats = saved_distributed_stats;
+      tile_nest[cur].replication_factor = saved_replication_factor;
     }
     else
     {