diff --git a/lib/substitutions/include/substitutions/operator_pattern.h b/lib/substitutions/include/substitutions/operator_pattern.h index 94e77cc51b..078a66a26d 100644 --- a/lib/substitutions/include/substitutions/operator_pattern.h +++ b/lib/substitutions/include/substitutions/operator_pattern.h @@ -82,6 +82,7 @@ enum class OperatorAttributeKey { NUM_INPUTS }; + /** * @brief OperatorAttributeValue is a representation of the concrete value of an attribute of an Operator. * The OperatorAttributeValue is evaluated from AttributeExpr. The datatype of the value corresponds to the @@ -121,7 +122,6 @@ using OperatorAttributeConstraint = using OperatorPattern = AttributePattern; - /** * @brief Given a specific attribute of an Operator, evaluate the expression of the attribute * using one of the three methods: direct value, list index access, or list size and return the diff --git a/lib/substitutions/include/substitutions/output_graph.h b/lib/substitutions/include/substitutions/output_graph.h index 72bce4f9af..2d1651dc66 100644 --- a/lib/substitutions/include/substitutions/output_graph.h +++ b/lib/substitutions/include/substitutions/output_graph.h @@ -27,6 +27,7 @@ struct AttrConstant { OperatorAttributeValue value; }; + /** * @brief OperatorAttributeExpr is a access to the attribute of an operator and can be * evaluated to a concrete value. OperatorAttributeExpr is used at substitution phase. diff --git a/lib/substitutions/include/substitutions/parallel_tensor_pattern.h b/lib/substitutions/include/substitutions/parallel_tensor_pattern.h index 786f2d2ff4..612486fe09 100644 --- a/lib/substitutions/include/substitutions/parallel_tensor_pattern.h +++ b/lib/substitutions/include/substitutions/parallel_tensor_pattern.h @@ -14,6 +14,7 @@ namespace FlexFlow { */ enum class TensorAttributeKey { DIM_SIZES, DIM_DEGREES }; + /** * @brief DIM_SIZES and DIM_DEGREES are represented by * a vector of ints that is listed as corresponding dimension @@ -34,6 +35,7 @@ using TensorAttributeConstraint = using ParallelTensorPattern = AttributePattern; + /** * @brief evaluate_attribute_expr evaluates the attribute expression for a given ParallelTensor * the ParallelTensor parameter is named tensor_shape because the numerical value will only be used