microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-24 14:54:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

add memory region functions

Browse Source
This commit is contained in:
Felipe Petroski Such
2023-04-07 12:26:56 -07:00
committed by Crutcher Dunnavant
parent 44a8a539ad
commit 38cd87cdcc
3 changed files with 112 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/include/ib.h
View File

@@ -12,21 +12,6 @@
#define MSCCLPP_IB_MAX_SENDS 64
#define MSCCLPP_IB_MAX_DEVS 8
// MR info to be shared with the remote peer
struct mscclppIbMrInfo
{
uint64_t addr;
uint32_t rkey;
};
// IB memory region
struct mscclppIbMr
{
struct ibv_mr* mr;
void* buff;
struct mscclppIbMrInfo info;
};
// QP info to be shared with the remote peer
struct mscclppIbQpInfo
{

30
src/include/mscclpp.h
View File

@@ -13,9 +13,11 @@
#include <mscclppfifo.h>
#include <time.h>
#include <vector>
#ifdef __cplusplus
extern "C" {
extern "C"
{
#endif
/***************************************************************************************************************
@@ -174,6 +176,32 @@ typedef struct
char internal[MSCCLPP_UNIQUE_ID_BYTES];
} mscclppUniqueId;
// MR info to be shared with the remote peer
struct mscclppIbMrInfo
{
uint64_t addr;
uint32_t rkey;
};
// IB memory region
struct mscclppIbMr
{
struct ibv_mr* mr;
void* buff;
struct mscclppIbMrInfo info;
};
struct mscclppRegisteredMemoryP2P
{
void* remoteBuff;
mscclppIbMr* IbMr;
};
struct mscclppRegisteredMemory
{
std::vector<mscclppRegisteredMemoryP2P> p2p;
};
/* Error type */
typedef enum
{