AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[GDA] implement rocshmem_p (#247)

Browse Source
This commit is contained in:
Yiltan
2025-09-11 09:24:43 -04:00
committed by GitHub
parent f677e5eb59
commit 2abeebbb6d
10 changed files with 75 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/gda/queue_pair.cpp
+7 -1
View File
@@ -446,7 +446,13 @@ __device__ void QueuePair::post_wqe_rma(int pe, int32_t size, uintptr_t *laddr,
SegmentBuilder seg_build(my_sq_index, sq_buf);
seg_build.update_ctrl_seg(my_sq_counter, opcode, 0, qp_num, MLX5_WQE_CTRL_CQ_UPDATE, 3, 0, 0);
seg_build.update_raddr_seg(raddr, rkey);
seg_build.update_data_seg(laddr, size, lkey);
if (size <= inline_threshold && opcode == GDA_OP_RDMA_WRITE) {
seg_build.update_inl_data_seg(laddr, size);
} else {
seg_build.update_data_seg(laddr, size, lkey);
}
__atomic_signal_fence(__ATOMIC_SEQ_CST);
if (is_leader) {