Skip to content

Commit 1ac9a4f

Browse files
committed
Test efa direct
Signed-off-by: Jessie Yang <[email protected]>
1 parent 9557aa6 commit 1ac9a4f

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

fabtests/benchmarks/benchmark_shared.c

+6-4
Original file line numberDiff line numberDiff line change
@@ -314,6 +314,7 @@ int pingpong_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
314314
return EXIT_FAILURE;
315315
}
316316

317+
inject_size = 0;
317318
/* Init rx_buf with invalid iteration number.
318319
* This must be done before the sender sends any data.
319320
*/
@@ -333,7 +334,7 @@ int pingpong_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
333334
if (rma_op == FT_RMA_WRITE)
334335
*(tx_buf + opts.transfer_size - 1) = (char)i;
335336

336-
if (opts.transfer_size <= inject_size)
337+
if (opts.transfer_size < inject_size)
337338
ret = ft_inject_rma(rma_op, remote, ep,
338339
remote_fi_addr,
339340
opts.transfer_size);
@@ -359,7 +360,7 @@ int pingpong_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
359360
if (rma_op == FT_RMA_WRITE)
360361
*(tx_buf + opts.transfer_size - 1) = (char)i;
361362

362-
if (opts.transfer_size <= inject_size)
363+
if (opts.transfer_size < inject_size)
363364
ret = ft_inject_rma(rma_op, remote, ep,
364365
remote_fi_addr,
365366
opts.transfer_size);
@@ -602,6 +603,7 @@ int bandwidth_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
602603
if (ft_check_opts(FT_OPT_VERIFY_DATA))
603604
inject_size = 0;
604605

606+
inject_size = 0;
605607
ret = ft_sync();
606608
if (ret)
607609
return ret;
@@ -629,7 +631,7 @@ int bandwidth_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
629631
}
630632
switch (rma_op) {
631633
case FT_RMA_WRITE:
632-
if (opts.transfer_size <= inject_size) {
634+
if (opts.transfer_size < inject_size) {
633635
ret = ft_post_rma_inject(FT_RMA_WRITE, tx_buf + offset,
634636
opts.transfer_size, remote);
635637
} else if (opts.use_fi_more) {
@@ -656,7 +658,7 @@ int bandwidth_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
656658
rx_seq++;
657659

658660
} else {
659-
if (opts.transfer_size <= inject_size) {
661+
if (opts.transfer_size < inject_size) {
660662
ret = ft_post_rma_inject(FT_RMA_WRITEDATA,
661663
tx_buf + offset,
662664
opts.transfer_size,

prov/efa/src/efa_domain.c

+2
Original file line numberDiff line numberDiff line change
@@ -283,6 +283,8 @@ int efa_domain_open(struct fid_fabric *fabric_fid, struct fi_info *info,
283283
goto err_free;
284284
}
285285
efa_domain->util_domain.domain_fid.ops = &efa_ops_domain_rdm;
286+
efa_domain->util_domain.domain_fid.ops->endpoint = efa_ep_open;
287+
efa_domain->util_domain.domain_fid.ops->cq_open = efa_cq_open;
286288
} else {
287289
assert(EFA_EP_TYPE_IS_DGRAM(info));
288290
efa_domain->util_domain.domain_fid.ops = &efa_ops_domain_dgram;

0 commit comments

Comments
 (0)