one directional push to expr group

Author: jjsjann123

csrc/val_graph.cpp

@@ -281,24 +281,29 @@ void ValGraph::initializeVal(
     const ExprGroup& expr_set =
         disjoint_exprs_.initializeSet(def).first->second;
     def_groups.pushBack(expr_set);
+
+    for (const auto& inp: def->inputs()) {
+      const ValGroup& inp_val_group = toGroup(inp);
+      unique_uses_[inp_val_group].pushBack(expr_set);
+    }
   }
   // TODO-NM: def_groups can be empty. Should it be still mapped?
   NVF_ERROR(
       unique_definitions_.emplace(val_disjoint_set, def_groups).second,
       "Multiple defining groups for ",
       nvfuser::toString(val_disjoint_set));
 
-  ExprGroups use_groups;
-  for (auto use : uses) {
-    const ExprGroup& expr_set =
-        disjoint_exprs_.initializeSet(use).first->second;
-    use_groups.pushBack(expr_set);
-  }
-  // TODO-NM: use_groups can be empty. Should it be still mapped?
-  NVF_ERROR(
-      unique_uses_.emplace(val_disjoint_set, use_groups).second,
-      "Multiple use groups for ",
-      nvfuser::toString(val_disjoint_set));
+  // ExprGroups use_groups;
+  // for (auto use : uses) {
+  //   const ExprGroup& expr_set =
+  //       disjoint_exprs_.initializeSet(use).first->second;
+  //   use_groups.pushBack(expr_set);
+  // }
+  // // TODO-NM: use_groups can be empty. Should it be still mapped?
+  // NVF_ERROR(
+  //     unique_uses_.emplace(val_disjoint_set, use_groups).second,
+  //     "Multiple use groups for ",
+  //     nvfuser::toString(val_disjoint_set));
 }
 
 void ValGraph::initializeVal(Val* val) {