Add numa-balanced mode to hpx::bind, spread cores over numa domains

STEllAR-GROUP · Sep 12, 2017 · 62f72e3 · 62f72e3
1 parent de6c7d7
commit 62f72e3
Show file tree

Hide file tree

Showing 2 changed files with 78 additions and 5 deletions.
diff --git a/hpx/runtime/threads/policies/parse_affinity_options.hpp b/hpx/runtime/threads/policies/parse_affinity_options.hpp
@@ -31,9 +31,10 @@ namespace hpx { namespace threads { namespace detail
 
     enum distribution_type
     {
-        compact  = 0x01,
-        scatter  = 0x02,
-        balanced = 0x04
+        compact       = 0x01,
+        scatter       = 0x02,
+        balanced      = 0x04,
+        numa_balanced = 0x08
     };
 
     struct spec_type

diff --git a/src/runtime/threads/policies/parse_affinity_options.cpp b/src/runtime/threads/policies/parse_affinity_options.cpp
@@ -117,6 +117,7 @@ namespace hpx { namespace threads { namespace detail
                     partlit("compact") >> qi::attr(compact)
                 |   partlit("scatter") >> qi::attr(scatter)
                 |   partlit("balanced") >> qi::attr(balanced)
+                |   partlit("numa-balanced") >> qi::attr(numa_balanced)
                 ;
 
             thread_spec =
@@ -818,9 +819,9 @@ namespace hpx { namespace threads { namespace detail
         // Iterate over the cores and assigned pus per core. this additional
         // loop is needed so that we have consecutive worker thread numbers
         std::size_t num_thread = 0;
-        for(std::size_t num_core = 0; num_core != num_cores; ++num_core)
+        for (std::size_t num_core = 0; num_core != num_cores; ++num_core)
         {
-            for(std::size_t num_pu = 0; num_pu != num_pus_cores[num_core]; ++num_pu)
+            for (std::size_t num_pu = 0; num_pu != num_pus_cores[num_core]; ++num_pu)
             {
                 if (any(affinities[num_thread]))
                 {
@@ -840,6 +841,72 @@ namespace hpx { namespace threads { namespace detail
         }
     }
 
+    ///////////////////////////////////////////////////////////////////////////
+    void decode_numabalanced_distribution(hwloc_topology_info& t,
+        std::vector<mask_type>& affinities,
+        std::size_t used_cores, std::size_t max_cores,
+        std::vector<std::size_t>& num_pus, error_code& ec)
+    {
+        std::size_t num_threads = affinities.size();
+        num_pus.resize(num_threads);
+        // numa nodes
+        std::size_t num_numas = (std::max)(std::size_t(1),t.get_number_of_numa_nodes());
+        std::vector<std::size_t> num_cores_numa(num_numas, 0);
+        std::vector<std::size_t> num_threads_numa(num_numas, 0);
+        std::size_t cores_t = 0;
+        for (std::size_t n=0; n<num_numas; ++n) {
+            num_cores_numa[n] = t.get_number_of_numa_node_cores(n);
+            cores_t += num_cores_numa[n];
+        }
+        // how many threads should go on each domain
+        for (std::size_t n=0; n<num_numas; ++n) {
+            num_threads_numa[n] = num_threads*num_cores_numa[n]/cores_t;
+        }
+
+        // assign threads to cores on each numa domain
+        std::size_t num_thread = 0;
+        std::size_t offset = 0;
+        for (std::size_t n = 0; n != num_numas; ++n)
+        {
+            std::vector<std::size_t> num_pus_cores(num_cores_numa[n], 0);
+
+            // iterate once and count pus/core
+            for (std::size_t thrd = 0; thrd != num_threads_numa[n]; /**/)
+            {
+                for(std::size_t c = 0; c != num_cores_numa[n]; ++c)
+                {
+                    num_pus_cores[c]++;
+                    if (++thrd == num_threads_numa[n])
+                        break;
+                }
+            }
+
+            // Iterate over the cores and assigned pus per core. this additional
+            // loop is needed so that we have consecutive worker thread numbers
+            for (std::size_t num_core = 0; num_core != num_cores_numa[n]; ++num_core)
+            {
+                for (std::size_t num_pu = 0; num_pu != num_pus_cores[num_core]; ++num_pu)
+                {
+                    if (any(affinities[num_thread]))
+                    {
+                        HPX_THROWS_IF(ec, bad_parameter,
+                            "decode_balancednuma_distribution",
+                            boost::str(boost::format(
+                                "affinity mask for thread %1% has "
+                                "already been set"
+                            ) % num_thread));
+                        return;
+                    }
+                    num_pus[num_thread] = t.get_pu_number(num_core + used_cores, num_pu);
+                    affinities[num_thread] = t.init_thread_affinity_mask(
+                        num_core + used_cores + offset, num_pu);
+                    ++num_thread;
+                }
+            }
+            offset += num_cores_numa[n];
+        }
+    }
+
     ///////////////////////////////////////////////////////////////////////////
     void decode_distribution(distribution_type d, hwloc_topology_info& t,
         std::vector<mask_type>& affinities,
@@ -863,6 +930,11 @@ namespace hpx { namespace threads { namespace detail
                 num_pus, ec);
             break;
 
+        case balanced_numa:
+            decode_numabalanced_distribution(t, affinities, used_cores, max_cores,
+                num_pus, ec);
+            break;
+
         default:
             HPX_ASSERT(false);
         }