-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #148 from mochi-hpc/carns/dev-abt-init-handling
improve handling of external Argobots initialization
- Loading branch information
Showing
4 changed files
with
250 additions
and
15 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,126 @@ | ||
|
|
||
| #include <margo.h> | ||
| #include "helper-server.h" | ||
| #include "munit/munit.h" | ||
|
|
||
| struct test_context { | ||
| margo_instance_id mid; | ||
| char* log_buffer; | ||
| int log_buffer_pos; | ||
| int log_buffer_size; | ||
| }; | ||
|
|
||
| static void test_log_fn(void* uargs, const char* str) | ||
| { | ||
| struct test_context* ctx = uargs; | ||
|
|
||
| /* check for overflow */ | ||
| munit_assert_int(strlen(str) + ctx->log_buffer_pos, <, | ||
| ctx->log_buffer_size); | ||
|
|
||
| /* directly copy msg to buffer */ | ||
| strcpy(&ctx->log_buffer[ctx->log_buffer_pos], str); | ||
| ctx->log_buffer_pos += strlen(str); | ||
|
|
||
| return; | ||
| } | ||
|
|
||
| struct margo_logger test_logger; | ||
|
|
||
| static void* test_context_setup(const MunitParameter params[], void* user_data) | ||
| { | ||
| (void)params; | ||
| (void)user_data; | ||
| int ret; | ||
|
|
||
| struct test_context* ctx = calloc(1, sizeof(*ctx)); | ||
| ctx->log_buffer = calloc(102400, 1); | ||
| ctx->log_buffer_size = 102400; | ||
|
|
||
| /* set up custom logger to make it easier to validate output */ | ||
| test_logger.uargs = ctx; | ||
| test_logger.trace = test_log_fn; | ||
| test_logger.debug = test_log_fn; | ||
| test_logger.info = test_log_fn; | ||
| test_logger.warning = test_log_fn; | ||
| test_logger.error = test_log_fn; | ||
| test_logger.critical = test_log_fn; | ||
|
|
||
| ret = margo_set_global_logger(&test_logger); | ||
| munit_assert_int(ret, ==, 0); | ||
|
|
||
| return ctx; | ||
| } | ||
|
|
||
| static void test_context_tear_down(void* data) | ||
| { | ||
| struct test_context* ctx = (struct test_context*)data; | ||
|
|
||
| free(ctx->log_buffer); | ||
| free(ctx); | ||
| } | ||
|
|
||
| static MunitResult margo_after_abt(const MunitParameter params[], void* data) | ||
| { | ||
| struct test_context* ctx = (struct test_context*)data; | ||
| char* protocol = "na+sm"; | ||
| int ret; | ||
|
|
||
| ret = ABT_init(0, NULL); | ||
| munit_assert_int(ret, ==, 0); | ||
|
|
||
| ctx->mid = margo_init(protocol, MARGO_CLIENT_MODE, 0, 0); | ||
| munit_assert_not_null(ctx->mid); | ||
|
|
||
| /* The above should have produced a warning, because margo was unable to | ||
| * set the desired abt stack settings | ||
| */ | ||
| munit_assert_int(ctx->log_buffer_pos, >, 0); | ||
| printf("global log contents: %s\n", ctx->log_buffer); | ||
|
|
||
| margo_finalize(ctx->mid); | ||
|
|
||
| return MUNIT_OK; | ||
| } | ||
|
|
||
| static MunitResult margo_after_abt_set_env(const MunitParameter params[], void* data) | ||
| { | ||
| struct test_context* ctx = (struct test_context*)data; | ||
| char* protocol = "na+sm"; | ||
| int ret; | ||
|
|
||
| /* In this version, we use a margo utility function to set desired | ||
| * parameters before calling ABT_init(). This should silence the | ||
| * warning. | ||
| */ | ||
| margo_set_environment(NULL); | ||
|
|
||
| ret = ABT_init(0, NULL); | ||
| munit_assert_int(ret, ==, 0); | ||
|
|
||
| ctx->mid = margo_init(protocol, MARGO_CLIENT_MODE, 0, 0); | ||
| munit_assert_not_null(ctx->mid); | ||
|
|
||
| /* check if log is silent */ | ||
| munit_assert_int(ctx->log_buffer_pos, ==, 0); | ||
|
|
||
| margo_finalize(ctx->mid); | ||
|
|
||
| return MUNIT_OK; | ||
| } | ||
|
|
||
|
|
||
| static MunitTest tests[] | ||
| = {{"/margo-after-abt", margo_after_abt, test_context_setup, | ||
| test_context_tear_down, MUNIT_TEST_OPTION_NONE, NULL}, | ||
| {"/margo-after-abt-set-env", margo_after_abt_set_env, test_context_setup, | ||
| test_context_tear_down, MUNIT_TEST_OPTION_NONE, NULL}, | ||
| {NULL, NULL, NULL, NULL, MUNIT_TEST_OPTION_NONE, NULL}}; | ||
|
|
||
| static const MunitSuite test_suite | ||
| = {"/margo", tests, NULL, 1, MUNIT_SUITE_OPTION_NONE}; | ||
|
|
||
| int main(int argc, char** argv) | ||
| { | ||
| return munit_suite_main(&test_suite, NULL, argc, argv); | ||
| } |