add umf split mode to os_memory_provider

split mode splits each allocation based on user provided partitions,
across multiple numa nodes. If no partition defined then each
allocation is split evenly, across all numa available.

Signed-off-by: Łukasz Plewa <[email protected]>

Author: lplewa

include/umf/providers/provider_os_memory.h

@@ -57,12 +57,23 @@ typedef enum umf_numa_mode_t {
     /// fulfilled, memory will be allocated from other nodes.
     UMF_NUMA_MODE_PREFERRED,
 
+    /// Allocation will be split evenly across nodes specified in nodemask.
+    /// An extra argument could be passed in params structure to specify other distribution.
+    UMF_NUMA_MODE_SPLIT,
     /// The memory is allocated on the node of the CPU that triggered the
     /// allocation. If this mode is specified, nodemask must be NULL and
     /// maxnode must be 0.
     UMF_NUMA_MODE_LOCAL, // TODO: should this be a hint or strict policy?
 } umf_numa_mode_t;
 
+/// @brief partition specification for UMF_NUMA_MODE_SPLIT
+typedef struct umf_numa_split_partition_t {
+    /// weight of partition.
+    unsigned weight;
+    /// numa node
+    unsigned target;
+} umf_numa_split_partition_t;
+
 /// @brief Memory provider settings struct
 typedef struct umf_os_memory_provider_params_t {
     /// Combination of 'umf_mem_protection_flags_t' flags
@@ -81,6 +92,11 @@ typedef struct umf_os_memory_provider_params_t {
     /// part size for interleave mode - 0 means default (system specific)
     /// It might be rounded up because of HW constraints
     size_t part_size;
+
+    /// ordered list of the partitions for split mode
+    umf_numa_split_partition_t *partitions;
+    /// len of partitions array
+    unsigned partitions_len;
 } umf_os_memory_provider_params_t;
 
 /// @brief OS Memory Provider operation results
@@ -106,8 +122,9 @@ umfOsMemoryProviderParamsDefault(void) {
         NULL,                                       /* numa_list */
         0,                                          /* numa_list_len */
         UMF_NUMA_MODE_DEFAULT,                      /* numa_mode */
-        0                                           /* part_size */
-    };
+        0,                                          /* part_size */
+        NULL,                                       /* partitions */
+        0};                                         /* partitions_len*/
 
     return params;
 }