mirror of
https://github.com/deepseek-ai/DeepEP
synced 2025-06-26 18:28:11 +00:00
Remove confusing comments
This commit is contained in:
parent
c4b8ffc37c
commit
ae0eafd2be
@ -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)
|
||||
|
||||
# Test cached dispatch (must without top-k staffs)
|
||||
# NOTES: handle must be refreshed
|
||||
if not with_topk:
|
||||
dispatch_args = {'x': current_x, 'handle': handle, 'config': config, 'async_finish': async_mode}
|
||||
if previous_mode:
|
||||
|
@ -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)
|
||||
|
||||
# Test cached dispatch (must without top-k staffs)
|
||||
# NOTES: handle must be refreshed
|
||||
if not with_topk:
|
||||
dispatch_args = {'x': current_x, 'handle': handle, 'config': config, 'async_finish': async_mode}
|
||||
if previous_mode:
|
||||
|
Loading…
Reference in New Issue
Block a user