diff --git a/apex/contrib/sparsity/permutation_lib.py b/apex/contrib/sparsity/permutation_lib.py index cc8e94bfc..c31f97df6 100644 --- a/apex/contrib/sparsity/permutation_lib.py +++ b/apex/contrib/sparsity/permutation_lib.py @@ -1200,6 +1200,7 @@ def fixup_concats(cls, fx_graph): fx_graph[child]['C_param'] = children_GCD_param old_children_GCD = cls.__group_data['sibling_group_C_params'][sibling_group_id] + children_GCD_param = str(np.gcd.reduce([int(children_GCD_param), int(old_children_GCD)])) cls.__group_data['sibling_group_C_params'][sibling_group_id] = children_GCD_param # fixup this node's dimensions