Skip to content

Retrieve cpuset when configured with pmix rte #8805

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
jsquyres merged 1 commit into from
Apr 19, 2021
Merged

Retrieve cpuset when configured with pmix rte #8805

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions ompi/mca/rte/pmix/rte_pmix.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
* and Technology (RIST). All rights reserved.
* Copyright (c) 2015 Intel, Inc. All rights reserved.
* Copyright (c) 2016-2017 IBM Corporation. All rights reserved.
* Copyright (c) 2021 Nanook Consulting. All rights reserved.
* $COPYRIGHT$
*
* Additional copyrights may follow
Expand Down Expand Up @@ -115,6 +116,7 @@ typedef struct {
int32_t num_local_peers;
uint32_t num_procs;
uint32_t app_num;
char *cpuset; /**< String-representation of bitmap where we are bound */
} pmix_process_info_t;
OMPI_DECLSPEC extern pmix_process_info_t pmix_process_info;
#define ompi_process_info pmix_process_info
Expand Down
12 changes: 12 additions & 0 deletions ompi/mca/rte/pmix/rte_pmix_module.c
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@
* of Tennessee Research Foundation. All rights
* reserved.
* Copyright (c) 2014 Cisco Systems, Inc. All rights reserved.
* Copyright (c) 2021 Nanook Consulting. All rights reserved.
* $COPYRIGHT$
*/
#include "ompi_config.h"
Expand Down Expand Up @@ -674,6 +675,17 @@ int ompi_rte_init(int *pargc, char ***pargv)
peers = NULL;
}

/* get our cpuset */
val = NULL;
OPAL_MODEX_RECV_VALUE_OPTIONAL(ret, OPAL_PMIX_CPUSET,
&pmix_process_info.my_name,
&val, OPAL_STRING);
if (OPAL_SUCCESS == ret && NULL != val) {
pmix_process_info.cpuset = val;
} else {
pmix_process_info.cpuset = NULL;
}

/* set the locality */
if (NULL != peers) {
/* identify our location */
Expand Down