From d3918584dd2b557735af63a31ab4da5d3e5c905e Mon Sep 17 00:00:00 2001 From: Giuseppe Franco Date: Thu, 23 Jan 2025 23:10:10 +0000 Subject: [PATCH] more fixes --- src/brevitas/graph/equalize.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/brevitas/graph/equalize.py b/src/brevitas/graph/equalize.py index b7d5dbaa7..4ab31ea80 100644 --- a/src/brevitas/graph/equalize.py +++ b/src/brevitas/graph/equalize.py @@ -1300,6 +1300,8 @@ def apply(self, alpha): co_optimize_act_weights=self.co_optimize_act_weights) scale_factors.append(scale_factor_region) rewriters.append(rewriters_region) + for r in rewriters_region: + r.apply(self.model) return scale_factors, rewriters_region def insert_mul_node(self, scale, shape, axis, act_node, batch_dim=0):