Skip to content

Commit 0fcac7a

Browse files
add mca parameters to limit partition size and count
Signed-off-by: Axel Schneewind <axel.schneewind@hlrs.de>
1 parent b257631 commit 0fcac7a

File tree

2 files changed

+22
-0
lines changed

2 files changed

+22
-0
lines changed

ompi/mca/part/persist_aggregated/part_persist_aggregated.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,10 @@ struct ompi_part_persist_aggregated_t {
8080
int32_t init_world;
8181
int32_t my_world_rank; /* Because the back end communicators use a world rank, we need to communicate ours
8282
to set up the requests. */
83+
84+
uint32_t min_message_size; /* parameters to control internal partitioning */
85+
uint32_t max_message_count;
86+
8387
opal_atomic_int32_t block_entry;
8488
opal_mutex_t lock;
8589
};

ompi/mca/part/persist_aggregated/part_persist_aggregated_component.c

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -88,6 +88,24 @@ mca_part_persist_aggregated_component_register(void)
8888
MCA_BASE_VAR_SCOPE_READONLY,
8989
&ompi_part_persist_aggregated.free_list_inc);
9090

91+
// variable for minimal internal partition size
92+
ompi_part_persist_aggregated.min_message_size = 4096;
93+
(void) mca_base_component_var_register(&mca_part_persist_aggregated_component.partm_version, "min_message_size",
94+
"Minimal size of transferred messages (internal partitions)",
95+
MCA_BASE_VAR_TYPE_UNSIGNED_INT, NULL, 0, 0,
96+
OPAL_INFO_LVL_9,
97+
MCA_BASE_VAR_SCOPE_READONLY,
98+
&ompi_part_persist_aggregated.min_message_size);
99+
100+
// variable for maximal internal partition count
101+
ompi_part_persist_aggregated.max_message_count = 4096;
102+
(void) mca_base_component_var_register(&mca_part_persist_aggregated_component.partm_version, "max_message_count",
103+
"Maximal number of transferred messages (internal partitions)",
104+
MCA_BASE_VAR_TYPE_UNSIGNED_INT, NULL, 0, 0,
105+
OPAL_INFO_LVL_9,
106+
MCA_BASE_VAR_SCOPE_READONLY,
107+
&ompi_part_persist_aggregated.max_message_count);
108+
91109

92110
return OPAL_SUCCESS;
93111
}

0 commit comments

Comments
 (0)