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 2374caf commit 340182bCopy full SHA for 340182b
tests/py/dynamo/distributed/test_nccl_ops.py
@@ -55,12 +55,7 @@ def setUpClass(cls):
55
print("USE_TRTLLM_PLUGINS =", os.environ.get("USE_TRTLLM_PLUGINS"))
56
cls.world_size = 1
57
if not dist.is_initialized():
58
- dist.init_process_group(
59
- backend="nccl",
60
- init_method="env://",
61
- world_size=cls.world_size,
62
- rank=0, # or read from env
63
- )
+ dist.init_process_group(backend="nccl")
64
cls.group = dist.new_group(ranks=[0])
65
cls.group_name = cls.group.group_name
66
0 commit comments