Skip to content

Commit b1b4422

Browse files
committed
upgrade hccl_buffer size
Signed-off-by: wangli <wangli858794774@gmail.com>
1 parent b1380f3 commit b1b4422

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

tests/e2e/multicard/test_data_parallel.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,8 @@
2727

2828
import pytest
2929

30+
os.environ["HCCL_BUFFSIZE"] = 500
31+
3032
MODELS = ["Qwen/Qwen3-0.6B", "Qwen/Qwen3-30B-A3B"]
3133

3234

tests/e2e/multicard/test_external_launcher.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@
2929
import pytest
3030
import torch_npu
3131

32+
os.environ["HCCL_BUFFSIZE"] = 500
33+
3234
MODELS = ["Qwen/Qwen3-0.6B"]
3335
MOE_MODELS = ["Qwen/Qwen3-30B-A3B"]
3436
DEVICE_NAME = torch_npu.npu.get_device_name(0)[:10]

0 commit comments

Comments
 (0)