@@ -9,9 +9,11 @@ def test_hierarchical_only(self):
99 af .examples .arches .networked .hierarchical_only ,
1010 )
1111 self .assertIn ("PeArray" , spec .arch .nodes )
12- self .assertEqual (spec .arch .nodes ["PeArray" ].get_fanout (), 4 )
12+ self .assertEqual (spec .arch .nodes ["PeArray" ].get_fanout (), 1 )
13+ self .assertIn ("Scratchpad" , spec .arch .nodes )
14+ self .assertEqual (spec .arch .nodes ["Scratchpad" ].get_fanout (), 4 )
1315 self .assertIn ("MacArray" , spec .arch .nodes )
14- self .assertEqual (spec .arch .nodes ["MacArray" ].get_fanout (), 4 )
16+ self .assertEqual (spec .arch .nodes ["MacArray" ].get_fanout (), 1 )
1517
1618 try :
1719 spec = spec .calculate_component_area_energy_latency_leak ()
@@ -23,7 +25,7 @@ def test_flat(self):
2325 af .examples .arches .networked .flat ,
2426 )
2527 self .assertIn ("NoC" , spec .arch .nodes )
26- self .assertEqual (spec .arch .nodes ["NoC" ].get_fanout (), 16 )
28+ self .assertEqual (spec .arch .nodes ["NoC" ].get_fanout (), 1 )
2729 self .assertEqual (
2830 {n .name for n in spec .arch .get_nodes_of_type (af .spec .Leaf )},
2931 {'MainMemory' , 'NoC' , 'GlobalBuffer' , 'RowBuffer' , 'ColumnBuffer' , 'DistributedBuffer' , 'Scratchpad' , 'MAC' }
0 commit comments