DeepSeek/DeepEP
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepEP synced 2025-05-31 18:38:18 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix style.

Browse Source
This commit is contained in:
Shangyan Zhou 2025-03-14 11:22:00 +08:00
parent 2d0cf41dd1
commit 38cdaf390c
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
csrc/kernels/ibgda_device.cuh
View File

@ -417,12 +417,11 @@ __device__ __forceinline__ void nvshmemi_ibgda_amo_nonfetch_add(void *rptr, cons
uint64_t raddr;
ibgda_get_rkey(reinterpret_cast<uint64_t>(rptr), pe, &raddr, &rkey);
void *wqe_ptrs[1];
uint64_t my_wqe_idx = ibgda_reserve_wqe_slots(qp, 1);
wqe_ptrs[0] = ibgda_get_wqe_ptr(qp, my_wqe_idx);
void *wqe_ptrs = ibgda_get_wqe_ptr(qp, my_wqe_idx);
ibgda_write_amo_add_wqe(qp, value, reinterpret_cast<uint64_t>(qp->ibuf.buf),
qp->ibuf.lkey, raddr, rkey, my_wqe_idx, wqe_ptrs);
qp->ibuf.lkey, raddr, rkey, my_wqe_idx, &wqe_ptrs);
ibgda_submit_requests<true>(qp, my_wqe_idx, 1);
}