Skip to content

Commit

Permalink
Providers/rxe: Add dma-buf support
Browse files Browse the repository at this point in the history
Implement a new provider method for dma-buf base memory registration.

Signed-off-by: Shunsuke Mie <[email protected]>
  • Loading branch information
ShunsukeMie committed Sep 30, 2021
1 parent c89d402 commit 09a6bd4
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions providers/rxe/rxe.c
Original file line number Diff line number Diff line change
Expand Up @@ -239,6 +239,26 @@ static struct ibv_mr *rxe_reg_mr(struct ibv_pd *pd, void *addr, size_t length,
return &vmr->ibv_mr;
}

static struct ibv_mr *rxe_reg_dmabuf_mr(struct ibv_pd *pd, uint64_t offset,
size_t length, uint64_t iova, int fd,
int access)
{
struct verbs_mr *vmr;
int ret;

vmr = malloc(sizeof(*vmr));
if (!vmr)
return NULL;

ret = ibv_cmd_reg_dmabuf_mr(pd, offset, length, iova, fd, access, vmr);
if (ret) {
free(vmr);
return NULL;
}

return &vmr->ibv_mr;
}

static int rxe_dereg_mr(struct verbs_mr *vmr)
{
int ret;
Expand Down Expand Up @@ -1706,6 +1726,7 @@ static const struct verbs_context_ops rxe_ctx_ops = {
.alloc_pd = rxe_alloc_pd,
.dealloc_pd = rxe_dealloc_pd,
.reg_mr = rxe_reg_mr,
.reg_dmabuf_mr = rxe_reg_dmabuf_mr,
.dereg_mr = rxe_dereg_mr,
.alloc_mw = rxe_alloc_mw,
.dealloc_mw = rxe_dealloc_mw,
Expand Down

0 comments on commit 09a6bd4

Please sign in to comment.