Test efa direct

Signed-off-by: Jessie Yang <[email protected]>
jiaxiyan · Jan 21, 2025 · c8a2119 · c8a2119
1 parent 71f34f0
commit c8a2119
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/fabtests/benchmarks/benchmark_shared.c b/fabtests/benchmarks/benchmark_shared.c
@@ -314,6 +314,7 @@ int pingpong_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
 		return EXIT_FAILURE;
 	}
 
+	inject_size = 0;
 	/* Init rx_buf with invalid iteration number.
 	 * This must be done before the sender sends any data.
 	 */
@@ -333,7 +334,7 @@ int pingpong_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
 			if (rma_op == FT_RMA_WRITE)
 				*(tx_buf + opts.transfer_size - 1) = (char)i;
 
-			if (opts.transfer_size <= inject_size)
+			if (opts.transfer_size < inject_size)
 				ret = ft_inject_rma(rma_op, remote, ep,
 						    remote_fi_addr,
 						    opts.transfer_size);
@@ -359,7 +360,7 @@ int pingpong_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
 			if (rma_op == FT_RMA_WRITE)
 				*(tx_buf + opts.transfer_size - 1) = (char)i;
 
-			if (opts.transfer_size <= inject_size)
+			if (opts.transfer_size < inject_size)
 				ret = ft_inject_rma(rma_op, remote, ep,
 						remote_fi_addr,
 						opts.transfer_size);
@@ -602,6 +603,7 @@ int bandwidth_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
 	if (ft_check_opts(FT_OPT_VERIFY_DATA))
 		inject_size = 0;
 
+	inject_size = 0;
 	ret = ft_sync();
 	if (ret)
 		return ret;
@@ -629,7 +631,7 @@ int bandwidth_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
 		}
 		switch (rma_op) {
 		case FT_RMA_WRITE:
-			if (opts.transfer_size <= inject_size) {
+			if (opts.transfer_size < inject_size) {
 				ret = ft_post_rma_inject(FT_RMA_WRITE, tx_buf + offset,
 						opts.transfer_size, remote);
 			} else if (opts.use_fi_more) {
@@ -656,7 +658,7 @@ int bandwidth_rma(enum ft_rma_opcodes rma_op, struct fi_rma_iov *remote)
 					rx_seq++;
 
 			} else {
-				if (opts.transfer_size <= inject_size) {
+				if (opts.transfer_size < inject_size) {
 					ret = ft_post_rma_inject(FT_RMA_WRITEDATA,
 							tx_buf + offset,
 							opts.transfer_size,

diff --git a/prov/efa/src/efa_domain.c b/prov/efa/src/efa_domain.c
@@ -283,6 +283,8 @@ int efa_domain_open(struct fid_fabric *fabric_fid, struct fi_info *info,
 			goto err_free;
 		}
 		efa_domain->util_domain.domain_fid.ops = &efa_ops_domain_rdm;
+		efa_domain->util_domain.domain_fid.ops->endpoint = efa_ep_open;
+		efa_domain->util_domain.domain_fid.ops->cq_open = efa_cq_open;
 	} else {
 		assert(EFA_EP_TYPE_IS_DGRAM(info));
 		efa_domain->util_domain.domain_fid.ops = &efa_ops_domain_dgram;