From 24453275e34b6c214615db74df4b47da614b8094 Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Wed, 18 Jun 2025 10:59:44 +0800 Subject: [PATCH] Add `EP_TEST_LL_COMPATIBILITY` --- tests/test_internode.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_internode.py b/tests/test_internode.py index 4aeca49..e84f4eb 100644 --- a/tests/test_internode.py +++ b/tests/test_internode.py @@ -220,7 +220,7 @@ def test_main(num_sms: int, local_rank: int, num_local_ranks: int, num_ranks: in def test_loop(local_rank: int, num_local_ranks: int): num_nodes = int(os.getenv('WORLD_SIZE', 1)) rank, num_ranks, group = init_dist(local_rank, num_local_ranks) - test_ll_compatibility = True + test_ll_compatibility = os.getenv('EP_TEST_LL_COMPATIBILITY', False) if test_ll_compatibility: ll_num_tokens, ll_hidden, ll_num_experts, ll_num_topk = 16, 5120, 256, 9