f

Author: AlexanderDokuchaev

tests/torch/data/reference_graphs/pruning_groups/Wave2Vec_2.0.dot

@@ -1,49 +1,49 @@
-strict digraph  {
-0 [color=red, label="Block: S:1__O:0
+strict digraph {
+0 [label="Block: S:1__O:0
 Producers: 19
-Consumers: ", style=filled];
-1 [color=grey, label="Block: S:1__O:0
+Consumers: ", color=red, style=filled];
+1 [label="Block: S:1__O:0
 Producers: 29
-Consumers: ", style=filled];
-2 [color=red, label="Block: S:24__O:0
+Consumers: ", color=grey, style=filled];
+2 [label="Block: S:24__O:0
 Producers: 29
-Consumers: ", style=filled];
-3 [color=red, label="Block: S:1__O:24
+Consumers: ", color=red, style=filled];
+3 [label="Block: S:1__O:24
 Producers: 29
-Consumers: ", style=filled];
-4 [color=grey, label="Block: S:1__O:0
-Producers: 30
-Consumers: ", style=filled];
-5 [color=red, label="Block: S:24__O:0
-Producers: 30
-Consumers: ", style=filled];
-6 [color=red, label="Block: S:1__O:24
-Producers: 30
-Consumers: ", style=filled];
-7 [color=grey, label="Block: S:1__O:0
-Producers: 34
-Consumers: ", style=filled];
-8 [color=red, label="Block: S:24__O:0
-Producers: 34
-Consumers: ", style=filled];
-9 [color=red, label="Block: S:1__O:24
-Producers: 34
-Consumers: ", style=filled];
-10 [color=red, label="Block: S:1__O:0
-Producers: 44
-Consumers: ", style=filled];
-11 [color=green, label="Block: S:1__O:0
-Producers: 48
-Consumers: 51", style=filled];
-12 [color=red, label="Block: S:1__O:0
-Producers: 51
-Consumers: ", style=filled];
-13 [color=red, label="Block: S:1__O:0
-Producers: 55
-Consumers: ", style=filled];
-14 [color=red, label="Block: S:1__O:0
+Consumers: ", color=red, style=filled];
+4 [label="Block: S:1__O:0
+Producers: 32
+Consumers: ", color=grey, style=filled];
+5 [label="Block: S:24__O:0
+Producers: 32
+Consumers: ", color=red, style=filled];
+6 [label="Block: S:1__O:24
+Producers: 32
+Consumers: ", color=red, style=filled];
+7 [label="Block: S:1__O:0
+Producers: 35
+Consumers: ", color=grey, style=filled];
+8 [label="Block: S:24__O:0
+Producers: 35
+Consumers: ", color=red, style=filled];
+9 [label="Block: S:1__O:24
+Producers: 35
+Consumers: ", color=red, style=filled];
+10 [label="Block: S:1__O:0
+Producers: 46
+Consumers: ", color=red, style=filled];
+11 [label="Block: S:1__O:0
+Producers: 50
+Consumers: 53", color=green, style=filled];
+12 [label="Block: S:1__O:0
+Producers: 53
+Consumers: ", color=red, style=filled];
+13 [label="Block: S:1__O:0
 Producers: 57
-Consumers: ", style=filled];
+Consumers: ", color=red, style=filled];
+14 [label="Block: S:1__O:0
+Producers: 59
+Consumers: ", color=red, style=filled];
 1 -> 2;
 1 -> 3;
 4 -> 5;

tests/torch/pruning/experimental/test_nodes_grouping.py

@@ -419,7 +419,7 @@ def __str__(self) -> str:
             ),
         ),
         ref_groups=[
-            PruningGroup(block=PruningBlock(), producers={ProducerInfo(48)}, consumers={ConsumerInfo(51)}),
+            PruningGroup(block=PruningBlock(), producers={ProducerInfo(50)}, consumers={ConsumerInfo(53)}),
         ],
     ),
 ]