Module Internal.Grouper

type cluster_t = {
  1. name : string;
  2. nodes : Ir.any_node list;
  3. inputs : Ir.Var.t list;
  4. outputs : Ir.Var.t list;
  5. vars_metadata : (Ir.Var.t, Ir.var_metadata) Hashtbl.t;
}
val get_node_input_vars : Ir.any_node -> Ir.Var.t list
val get_node_output_var : Ir.any_node -> Ir.Var.t
val is_boundary_node : Ir.any_node -> bool
val is_fusible_elementwise : Ir.any_node -> bool
val group : Ir.graph_t -> cluster_t list