From ae0eafd2be0ad471f58f53d1b029b9fd6602dd1e Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Tue, 25 Mar 2025 09:27:34 +0800 Subject: [PATCH] Remove confusing comments --- tests/test_internode.py | 1 - tests/test_intranode.py | 1 - 2 files changed, 2 deletions(-) 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: