Commit ae0eafd2 authored by Chenggang Zhao's avatar Chenggang Zhao
Browse files

Remove confusing comments

parent c4b8ffc3
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment