Skip to content

Commit 1a5aacc

Browse files
committed
enables using progress thread
1 parent 154db67 commit 1a5aacc

File tree

2 files changed

+14
-4
lines changed

2 files changed

+14
-4
lines changed

benchmark/allreduce_benchmark.c

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@ typedef struct options_t {
1818
char* transport;
1919
unsigned iterations;
2020
unsigned msg_size;
21+
na_bool_t use_progress_thread;
2122
} options_t;
2223

2324
static void run_mpi_benchmark(options_t* options) {
@@ -67,7 +68,7 @@ static void run_mona_benchmark(options_t* options) {
6768
MPI_Comm_size(MPI_COMM_WORLD, &size);
6869
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
6970

70-
mona_instance_t mona = mona_init_thread(options->transport, NA_TRUE, NULL, NA_FALSE);
71+
mona_instance_t mona = mona_init_thread(options->transport, NA_TRUE, NULL, options->use_progress_thread);
7172
ASSERT_MESSAGE(mona != MONA_INSTANCE_NULL, "Could not initialize Mona instance");
7273

7374
char addr_str[128];
@@ -138,8 +139,9 @@ static void parse_options(int argc, char** argv, options_t* options) {
138139
options->msg_size = 128;
139140
options->transport = (char*)default_transport;
140141
options->method = NULL;
142+
options->use_progress_thread = NA_FALSE;
141143

142-
while((c = getopt(argc, argv, "i:s:m:t:h")) != -1) {
144+
while((c = getopt(argc, argv, "i:s:m:t:hp")) != -1) {
143145
switch (c)
144146
{
145147
case 'h':
@@ -158,6 +160,9 @@ static void parse_options(int argc, char** argv, options_t* options) {
158160
case 't':
159161
options->transport = optarg;
160162
break;
163+
case 'p':
164+
options->use_progress_thread = NA_TRUE;
165+
break;
161166
case '?':
162167
if(optopt == 'i' || optopt == 's' || optopt == 'm')
163168
fprintf(stderr, "Option -%c requires an argument.\n", optopt);

benchmark/send_recv_benchmark.c

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ typedef struct options_t {
1717
char* transport;
1818
unsigned iterations;
1919
unsigned msg_size;
20+
na_bool_t use_progress_thread;
2021
} options_t;
2122

2223
static void run_mpi_benchmark(options_t* options) {
@@ -56,7 +57,7 @@ static void run_mona_benchmark(options_t* options) {
5657
unsigned i;
5758
MPI_Comm_rank(MPI_COMM_WORLD, &rank);
5859

59-
mona_instance_t mona = mona_init_thread(options->transport, NA_TRUE, NULL, NA_FALSE);
60+
mona_instance_t mona = mona_init_thread(options->transport, NA_TRUE, NULL, options->use_progress_thread);
6061
ASSERT_MESSAGE(mona != MONA_INSTANCE_NULL, "Could not initialize Mona instance");
6162

6263
char addr_str[128];
@@ -116,8 +117,9 @@ static void parse_options(int argc, char** argv, options_t* options) {
116117
options->msg_size = 128;
117118
options->transport = (char*)default_transport;
118119
options->method = NULL;
120+
options->use_progress_thread = NA_FALSE;
119121

120-
while((c = getopt(argc, argv, "i:s:m:t:")) != -1) {
122+
while((c = getopt(argc, argv, "i:s:m:t:p")) != -1) {
121123
switch (c)
122124
{
123125
case 'i':
@@ -132,6 +134,9 @@ static void parse_options(int argc, char** argv, options_t* options) {
132134
case 't':
133135
options->transport = optarg;
134136
break;
137+
case 'p':
138+
options->use_progress_thread = NA_TRUE;
139+
break;
135140
case '?':
136141
if(optopt == 'i' || optopt == 's' || optopt == 'm')
137142
fprintf(stderr, "Option -%c requires an argument.\n", optopt);

0 commit comments

Comments
 (0)