[fastfusion] model bugfixes

Michael Gilbert · Michael Gilbert · commit 9821548f66d2 · 2025-01-29T16:23:00.000-05:00
diff --git a/pytimeloop/fastfusion/fastmodel/fastmodel.py b/pytimeloop/fastfusion/fastmodel/fastmodel.py
@@ -179,6 +179,8 @@ def compile_mapping(mapping,
 
                 actual_tensor_access_multiplier[tensor_id] *= \
                     fill_multicast_factor[tensor_id]
+                potential_tensor_access_multiplier[tensor_id] *= \
+                    fill_multicast_factor[tensor_id]
                 fill_multicast_factor[tensor_id] = 1
 
                 if target not in fanout:
