Comment

Author: wujingyue

csrc/host_ir/lowering.cpp

@@ -174,7 +174,8 @@ void lowerSegment(
       // without cloning the value again.
       Expr* e = ir_cloner.clone(group.exprs().front());
 
-      // FIXME: should this be associated with the scope?
+      // TODO: `replacement_map` should be associated with the scope os
+      // ShardByStream across segments in the same for-loop can be reused.
       std::unordered_map<Val*, Val*> replacement_map;
       for (Expr* c : convertSingleOpToCommunication(e, device_id)) {
         NVF_ERROR(