Skip to content

Commit 2828bc7

Browse files
authored
fix ut (#10812)
1 parent 9db95ee commit 2828bc7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/utils/test_nccl_config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def test_nccl_config(self):
6060
strategy = fleet.DistributedStrategy()
6161
strategy.hybrid_configs = {"dp_degree": 2, "mp_degree": 2, "pp_degree": 2}
6262

63-
strategy = init_nccl_config(strategy)
63+
strategy = init_nccl_config(self.nccl_config_path, strategy)
6464
assert strategy.hybrid_configs["default_comm_group_configs"].nccl_config.protoStr == "ll"
6565
assert strategy.hybrid_configs["default_comm_group_configs"].nccl_config.nchannels == 1
6666
assert strategy.hybrid_configs["default_comm_group_configs"].nccl_config.buffsize_align == 1024

0 commit comments

Comments
 (0)