Format

Author: lockshaw

lib/substitutions/src/substitutions/apply_substitution/perform_shape_inference.cc

@@ -1,8 +1,10 @@
 #include "substitutions/apply_substitution/perform_shape_inference.h"
 #include "op-attrs/get_incoming_tensor_roles.h"
 #include "op-attrs/shape_inference.h"
+#include "utils/containers/binary_merge_disjoint_maps.h"
 #include "utils/containers/filter_values.h"
 #include "utils/containers/filtrans.h"
+#include "utils/containers/is_subseteq_of.h"
 #include "utils/containers/map_keys.h"
 #include "utils/containers/map_values.h"
 #include "utils/containers/restrict_keys.h"
@@ -18,8 +20,6 @@
 #include "utils/graph/open_dataflow_graph/algorithms/get_inputs.h"
 #include "utils/graph/open_kwarg_dataflow_graph/algorithms/get_incoming_open_kwarg_dataflow_values_for_node.h"
 #include "utils/nonnegative_int/num_elements.h"
-#include "utils/containers/binary_merge_disjoint_maps.h"
-#include "utils/containers/is_subseteq_of.h"
 
 namespace FlexFlow {
 
@@ -72,7 +72,8 @@ LabelledOpenKwargDataflowGraphView<ParallelLayerAttrs,
     std::unordered_map<TensorSlotName, ParallelTensorShape> weight_shapes =
         incoming_shapes_with_role(IncomingTensorRole::WEIGHT);
 
-    ASSERT(binary_merge_disjoint_maps(input_shapes, weight_shapes) == incoming_shapes);
+    ASSERT(binary_merge_disjoint_maps(input_shapes, weight_shapes) ==
+           incoming_shapes);
 
     std::unordered_map<TensorSlotName, ParallelTensorShape>
         inferred_weight_shapes =