From 853d560e6fda18e8ec3081a0f3e5a88d26a42fb7 Mon Sep 17 00:00:00 2001 From: unknown Date: Fri, 21 Jun 2024 18:16:05 +0800 Subject: [PATCH] fixup concats for grouped convolution --- apex/contrib/sparsity/permutation_lib.py | 1 + 1 file changed, 1 insertion(+) 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