Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
DeepEP
Commits
58c47942
"...ssh:/git@developer.sourcefind.cn:2222/tsoc/openmm.git" did not exist on "1ec69c9544f138cf2ec5563d6d5b734d8eefcf35"
Commit
58c47942
authored
Jun 27, 2025
by
Shangyan Zhou
Browse files
Minor fix.
parent
7ce8da4e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
2 deletions
+1
-2
csrc/kernels/internode.cu
csrc/kernels/internode.cu
+1
-2
No files found.
csrc/kernels/internode.cu
View file @
58c47942
...
@@ -560,7 +560,6 @@ dispatch(int4* recv_x, float* recv_x_scales, int64_t* recv_topk_idx, float* recv
...
@@ -560,7 +560,6 @@ dispatch(int4* recv_x, float* recv_x_scales, int64_t* recv_topk_idx, float* recv
if
(
is_token_in_rank_uint64
!=
0
)
{
if
(
is_token_in_rank_uint64
!=
0
)
{
// Acquire lock first
// Acquire lock first
acquire_lock
(
rdma_send_channel_lock
+
lane_id
);
acquire_lock
(
rdma_send_channel_lock
+
lane_id
);
auto
window
=
rdma_send_channel_window
[
lane_id
];
auto
latest_tail
=
rdma_send_channel_tail
[
lane_id
];
auto
latest_tail
=
rdma_send_channel_tail
[
lane_id
];
auto
offset
=
rdma_tail_idx
-
latest_tail
;
auto
offset
=
rdma_tail_idx
-
latest_tail
;
while
(
offset
>=
32
)
{
while
(
offset
>=
32
)
{
...
@@ -572,7 +571,7 @@ dispatch(int4* recv_x, float* recv_x_scales, int64_t* recv_topk_idx, float* recv
...
@@ -572,7 +571,7 @@ dispatch(int4* recv_x, float* recv_x_scales, int64_t* recv_topk_idx, float* recv
// Release the transaction slot
// Release the transaction slot
// Add the bit and move the ones if possible
// Add the bit and move the ones if possible
window
|
=
1u
<<
offset
;
auto
window
=
rdma_send_channel_window
[
lane_id
]
|
(
1u
<<
offset
)
;
if
(
offset
==
0
)
{
if
(
offset
==
0
)
{
auto
num_empty_slots
=
(
~
window
)
==
0
?
32
:
__ffs
(
~
window
)
-
1
;
auto
num_empty_slots
=
(
~
window
)
==
0
?
32
:
__ffs
(
~
window
)
-
1
;
st_release_cta
(
rdma_send_channel_tail
+
lane_id
,
latest_tail
+
num_empty_slots
);
st_release_cta
(
rdma_send_channel_tail
+
lane_id
,
latest_tail
+
num_empty_slots
);
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment