wip

Author: jjsjann123

csrc/val_graph.cpp

@@ -284,7 +284,7 @@ void ValGraph::initializeVal(
 
     for (const auto& inp: def->inputs()) {
       const ValGroup& inp_val_group = toGroup(inp);
-      unique_uses_[inp_val_group].pushBack(expr_set);
+      unique_uses_.at(inp_val_group).pushBack(expr_set);
     }
   }
   // TODO-NM: def_groups can be empty. Should it be still mapped?
@@ -293,6 +293,10 @@ void ValGraph::initializeVal(
       "Multiple defining groups for ",
       nvfuser::toString(val_disjoint_set));
 
+  NVF_ERROR(
+     unique_uses_.emplace(val_disjoint_set, {}).second,
+     "Multiple use groups for ",
+     nvfuser::toString(val_disjoint_set));
   // ExprGroups use_groups;
   // for (auto use : uses) {
   //   const ExprGroup& expr_set =