diff --git a/tests/test_internode.py b/tests/test_internode.py index 73c3bbd..765ba1a 100644 --- a/tests/test_internode.py +++ b/tests/test_internode.py @@ -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: diff --git a/tests/test_intranode.py b/tests/test_intranode.py index 1c02751..107fea4 100644 --- a/tests/test_intranode.py +++ b/tests/test_intranode.py @@ -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: