We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9db95ee commit 2828bc7Copy full SHA for 2828bc7
tests/utils/test_nccl_config.py
@@ -60,7 +60,7 @@ def test_nccl_config(self):
60
strategy = fleet.DistributedStrategy()
61
strategy.hybrid_configs = {"dp_degree": 2, "mp_degree": 2, "pp_degree": 2}
62
63
- strategy = init_nccl_config(strategy)
+ strategy = init_nccl_config(self.nccl_config_path, strategy)
64
assert strategy.hybrid_configs["default_comm_group_configs"].nccl_config.protoStr == "ll"
65
assert strategy.hybrid_configs["default_comm_group_configs"].nccl_config.nchannels == 1
66
assert strategy.hybrid_configs["default_comm_group_configs"].nccl_config.buffsize_align == 1024
0 commit comments