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

Remove confusing comments

Browse Source
This commit is contained in:
Chenggang Zhao 2025-03-25 09:27:34 +08:00
parent c4b8ffc37c
commit ae0eafd2be
2 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_internode.py
View File

@ -128,7 +128,6 @@ def test_main(num_sms: int, local_rank: int, num_local_ranks: int, num_ranks: in
check_data(recv_topk_weights, recv_gbl_rank_prefix_sum) check_data(recv_topk_weights, recv_gbl_rank_prefix_sum)
# Test cached dispatch (must without top-k staffs) # Test cached dispatch (must without top-k staffs)
# NOTES: handle must be refreshed
if not with_topk: if not with_topk:
dispatch_args = {'x': current_x, 'handle': handle, 'config': config, 'async_finish': async_mode} dispatch_args = {'x': current_x, 'handle': handle, 'config': config, 'async_finish': async_mode}
if previous_mode: if previous_mode:

1
tests/test_intranode.py
View File

@ -112,7 +112,6 @@ def test_main(num_sms: int, local_rank: int, num_ranks: int, rank: int, buffer:
check_data(recv_topk_weights, rank_prefix_matrix) check_data(recv_topk_weights, rank_prefix_matrix)
# Test cached dispatch (must without top-k staffs) # Test cached dispatch (must without top-k staffs)
# NOTES: handle must be refreshed
if not with_topk: if not with_topk:
dispatch_args = {'x': current_x, 'handle': handle, 'config': config, 'async_finish': async_mode} dispatch_args = {'x': current_x, 'handle': handle, 'config': config, 'async_finish': async_mode}
if previous_mode: if previous_mode: