Skip to content

Commit 68a7310

Browse files
wckzhangPalmer Stolly
andcommitted
mtl/ofi: Update structs for FI_HMEM support
Update MTL/OFI structs to store needed data. Signed-off-by: William Zhang <wilzhang@amazon.com> Co-authored-by: William Zhang <wilzhang@amazon.com> Co-authored-by: Palmer Stolly <pstolly@amazon.com>
1 parent 249420f commit 68a7310

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

ompi/mca/mtl/ofi/mtl_ofi_request.h

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,10 @@ struct ompi_mtl_ofi_request_t {
8787

8888
/** Pointer to Mrecv request to complete */
8989
struct mca_mtl_request_t *mrecv_req;
90+
91+
/** Stores reference to memory region from registration */
92+
/* Set to NULL if memory not registered or if non CUDA buffer */
93+
struct fid_mr *mr;
9094
};
9195
typedef struct ompi_mtl_ofi_request_t ompi_mtl_ofi_request_t;
9296

0 commit comments

Comments
 (0)