Skip to content
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

Jump to bottom

btl/uct: add support for using an another memory domain to form connections #12822

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
+475 −237
Open

btl/uct: add support for using an another memory domain to form connections #12822

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
4dc817c
btl/uct: add support for using an another memory domain to form conne…
hjelmn Sep 13, 2024
db2f692
Merge branch 'main' into update_btl_uct_to_support_alternate_means_of…
hjelmn Sep 23, 2024
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
23 changes: 22 additions & 1 deletion opal/mca/btl/uct/btl_uct.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -139,9 +139,15 @@ struct mca_btl_uct_component_t {

/** allowed UCT memory domains */
char *memory_domains;
mca_btl_uct_include_list_t memory_domain_list;

/** allowed transports */
char *allowed_transports;
mca_btl_uct_include_list_t allowed_transport_list;

/** transports to consider for forming connections */
char *connection_domains;
mca_btl_uct_include_list_t connection_domain_list;

/** number of worker contexts to create */
int num_contexts_per_module;
Expand All @@ -153,6 +159,10 @@ struct mca_btl_uct_component_t {

/** disable UCX memory hooks */
bool disable_ucx_memory_hooks;

/** alternate connection-only module that can be used if no suitable
* connection tl is found. this is usually a tcp tl. */
mca_btl_uct_module_t *conn_module;
};
typedef struct mca_btl_uct_component_t mca_btl_uct_component_t;

Expand Down Expand Up @@ -289,7 +299,8 @@ struct mca_btl_base_endpoint_t *mca_btl_uct_get_ep(struct mca_btl_base_module_t
opal_proc_t *proc);

int mca_btl_uct_query_tls(mca_btl_uct_module_t *module, mca_btl_uct_md_t *md,
uct_tl_resource_desc_t *tl_descs, unsigned tl_count);
uct_tl_resource_desc_t *tl_descs, unsigned tl_count,
bool evaluate_for_conn_only);
int mca_btl_uct_process_connection_request(mca_btl_uct_module_t *module,
mca_btl_uct_conn_req_t *req);

Expand Down Expand Up @@ -336,5 +347,15 @@ static inline bool mca_btl_uct_tl_requires_connection_tl(mca_btl_uct_tl_t *tl)
return !(MCA_BTL_UCT_TL_ATTR(tl, 0).cap.flags & UCT_IFACE_FLAG_CONNECT_TO_IFACE);
}

/**
* @brief Find the rank of `name` in the include list `list`.
*
* @param[in] name name to find
* @param[in] list list to search
*
* A negative result means the name is not present or the list is negated.
*/
int mca_btl_uct_include_list_rank (const char *name, const mca_btl_uct_include_list_t *list);

END_C_DECLS
#endif
Loading
Loading