vtr_flow: added script to add tiles to architecture xml

Signed-off-by: Alessandro Comodi <[email protected]>

Author: acomodi

vtr_flow/scripts/add_tiles.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+
+"""
+This script is intended to modify the architecture description file to be compliant with
+the new format.
+
+It moves the top level pb_types attributes and tags to the tiles high-level tag.
+
+BEFORE:
+<complexblocklist>
+    <pb_type name="BRAM" area="2" height="4" width="1" capacity="1">
+        <input ... />
+        <input ... />
+        <input ... />
+        <output ... />
+        <output ... />
+        <output ... />
+        <interconnect ... />
+        <fc ... />
+        <pinlocations ... />
+        <switchblock_locations ... />
+    </pb_type>
+</complexblocklist>
+
+AFTER:
+<tiles>
+    <tile name="BRAM" area="2" height="4" width="1" capacity="1">
+        <interconnect ... />
+        <fc ... />
+        <pinlocations ... />
+        <switchblock_locations ... />
+    </tile>
+</tiles>
+<complexblocklist
+    <pb_type name="BRAM">
+        <input ... />
+        <input ... />
+        <input ... />
+        <output ... />
+        <output ... />
+        <output ... />
+    </pb_type>
+</complexblocklist>
+"""
+
+from lxml import etree as ET
+import argparse
+
+TAGS_TO_SWAP = ['fc', 'pinlocations', 'switchblock_locations']
+ATTR_TO_REMOVE = ['area', 'height', 'width', 'capacity']
+
+def swap_tags(tile, pb_type):
+    # Moving tags from top level pb_type to tile
+    for child in pb_type:
+        if child.tag in TAGS_TO_SWAP:
+            pb_type.remove(child)
+            tile.append(child)
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Moves top level pb_types to tiles tag."
+    )
+    parser.add_argument(
+        '--arch_xml',
+        required=True,
+        help="Input arch.xml that needs to be modified to move the top level pb_types to the `tiles` tag."
+    )
+
+    args = parser.parse_args()
+
+    arch_xml = ET.ElementTree()
+    root_element = arch_xml.parse(args.arch_xml)
+
+    tiles = ET.SubElement(root_element, 'tiles')
+
+    top_pb_types = []
+    for pb_type in root_element.iter('pb_type'):
+        if pb_type.getparent().tag == 'complexblocklist':
+            top_pb_types.append(pb_type)
+
+    for pb_type in top_pb_types:
+        tile = ET.SubElement(tiles, 'tile')
+        attrs = pb_type.attrib
+
+        for attr in attrs:
+            tile.set(attr, pb_type.get(attr))
+
+        # Remove attributes of top level pb_types only
+        for attr in ATTR_TO_REMOVE:
+            pb_type.attrib.pop(attr, None)
+
+        swap_tags(tile, pb_type)
+
+    print(ET.tostring(arch_xml, pretty_print=True).decode('utf-8'))
+
+
+if __name__ == '__main__':
+    main()

vtr_flow/scripts/run_vtr_flow.pl

@@ -366,7 +366,7 @@
 
 # Read arch XML
 my $tpp      = XML::TreePP->new();
-my $xml_tree = $tpp->parsefile($architecture_file_path);
+my $xml_tree = $tpp->parsefile("$architecture_file_path");
 
 # Get lut size if undefined
 if (!defined $lut_size) {
@@ -419,7 +419,8 @@
 #system "cp $odin2_base_config"
 
 my $architecture_file_path_new = "$temp_dir$architecture_file_name";
-copy( $architecture_file_path, $architecture_file_path_new );
+my $ret = `$vtr_flow_path/scripts/add_tiles.py --arch_xml $architecture_file_path > $architecture_file_path_new`;
+#copy( "$architecture_file_path", $architecture_file_path_new );
 $architecture_file_path = $architecture_file_path_new;
 
 my $circuit_file_path_new = "$temp_dir$benchmark_name" . file_ext_for_stage($starting_stage - 1, $circuit_suffix);