Added command line option to turn attraction groups on or off

Author: sfkhalid

vpr/src/base/SetupVPR.cpp

@@ -495,6 +495,7 @@ void SetupPackerOpts(const t_options& Options,
     PackerOpts->high_fanout_threshold = Options.pack_high_fanout_threshold;
     PackerOpts->transitive_fanout_threshold = Options.pack_transitive_fanout_threshold;
     PackerOpts->feasible_block_array_size = Options.pack_feasible_block_array_size;
+    PackerOpts->use_attraction_groups = Options.use_attraction_groups;
 
     //TODO: document?
     PackerOpts->inter_cluster_net_delay = 1.0; /* DEFAULT */

vpr/src/base/read_options.cpp

@@ -1638,6 +1638,11 @@ argparse::ArgumentParser create_arg_parser(std::string prog_name, t_options& arg
         .default_value("2")
         .show_in(argparse::ShowIn::HELP_ONLY);
 
+    pack_grp.add_argument<bool, ParseOnOff>(args.use_attraction_groups, "--use_attraction_groups")
+        .help("Whether attraction groups are used to make it easier to pack primitives in the same floorplan region together.")
+        .default_value("on")
+        .show_in(argparse::ShowIn::HELP_ONLY);
+
     auto& place_grp = parser.add_argument_group("placement options");
 
     place_grp.add_argument(args.Seed, "--seed")

vpr/src/base/read_options.h

@@ -91,6 +91,7 @@ struct t_options {
     argparse::ArgValue<int> pack_feasible_block_array_size;
     argparse::ArgValue<std::vector<std::string>> pack_high_fanout_threshold;
     argparse::ArgValue<int> pack_verbosity;
+    argparse::ArgValue<bool> use_attraction_groups;
 
     /* Placement options */
     argparse::ArgValue<int> Seed;

vpr/src/base/vpr_types.h

@@ -814,6 +814,7 @@ struct t_packer_opts {
     enum e_packer_algorithm packer_algorithm;
     std::string device_layout;
     e_timing_update_type timing_update_type;
+    bool use_attraction_groups;
 };
 
 /**

vpr/src/pack/attraction_groups.cpp

@@ -1,37 +1,39 @@
 #include "attraction_groups.h"
 
-AttractionInfo::AttractionInfo() {
+AttractionInfo::AttractionInfo(bool attraction_groups_on) {
     auto& floorplanning_ctx = g_vpr_ctx.mutable_floorplanning();
     auto& atom_ctx = g_vpr_ctx.atom();
     int num_parts = floorplanning_ctx.constraints.get_num_partitions();
 
-    //Create an attraction group for each partition in the floorplanning constraints
-    for (int ipart = 0; ipart < num_parts; ipart++) {
-        PartitionId partid(ipart);
-        Partition part = floorplanning_ctx.constraints.get_partition(partid);
-
-        AttractionGroup group_info;
-        group_info.group_atoms = floorplanning_ctx.constraints.get_part_atoms(partid);
-
-        attraction_groups.push_back(group_info);
-    }
-
     //Initialize every atom to have no attraction group id
     int num_atoms = atom_ctx.nlist.blocks().size();
 
     atom_attraction_group.resize(num_atoms);
     fill(atom_attraction_group.begin(), atom_attraction_group.end(), AttractGroupId::INVALID());
 
-    //Then, fill in the group id for the atoms that do have an attraction group
-    int num_att_grps = attraction_groups.size();
+    //Create an attraction group for each partition in the floorplanning constraints
+    if (attraction_groups_on) {
+        for (int ipart = 0; ipart < num_parts; ipart++) {
+            PartitionId partid(ipart);
+            Partition part = floorplanning_ctx.constraints.get_partition(partid);
+
+            AttractionGroup group_info;
+            group_info.group_atoms = floorplanning_ctx.constraints.get_part_atoms(partid);
+
+            attraction_groups.push_back(group_info);
+        }
+
+        //Then, fill in the group id for the atoms that do have an attraction group
+        int num_att_grps = attraction_groups.size();
 
-    for (int igroup = 0; igroup < num_att_grps; igroup++) {
-        AttractGroupId group_id(igroup);
+        for (int igroup = 0; igroup < num_att_grps; igroup++) {
+            AttractGroupId group_id(igroup);
 
-        AttractionGroup att_group = attraction_groups[group_id];
+            AttractionGroup att_group = attraction_groups[group_id];
 
-        for (unsigned int iatom = 0; iatom < att_group.group_atoms.size(); iatom++) {
-            atom_attraction_group[att_group.group_atoms[iatom]] = group_id;
+            for (unsigned int iatom = 0; iatom < att_group.group_atoms.size(); iatom++) {
+                atom_attraction_group[att_group.group_atoms[iatom]] = group_id;
+            }
         }
     }
 }

vpr/src/pack/attraction_groups.h

@@ -41,7 +41,7 @@ struct AttractionGroup {
      * Atoms belonging to this attraction group will receive this gain if they
      * are potential candidates to be put in a cluster with the same attraction group.
      */
-    float gain = 3;
+    float gain = 7;
 
     /*
      * If the group is made up from a partition of atoms that are confined to a size one spot
@@ -56,7 +56,7 @@ class AttractionInfo {
   public:
     //Constructor that fills in the attraction groups based on vpr's floorplan constraints.
     //If no constraints were specified, then no attraction groups will be created.
-    AttractionInfo();
+    AttractionInfo(bool attraction_groups_on);
 
     //Setters and getters for the class
     AttractGroupId get_atom_attraction_group(const AtomBlockId atom_id);

vpr/src/pack/pack.cpp

@@ -100,7 +100,7 @@ bool try_pack(t_packer_opts* packer_opts,
                                                                expected_lowest_cost_pb_gnode,
                                                                list_of_packing_patterns.size()));
 
-    AttractionInfo attraction_groups;
+    AttractionInfo attraction_groups(packer_opts->use_attraction_groups);
     VTR_LOG("%d attraction groups were created during prepacking.\n", attraction_groups.num_attraction_groups());
     VTR_LOG("Finish prepacking.\n");