Searched refs:distributed_backend (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/distributed/ddp/ |
| H A D | benchmark.py | 84 group = dist.new_group(ranks=ranks, backend=benchmark.distributed_backend)
153 def __init__(self, device, distributed_backend, bucket_size): argument
156 self.distributed_backend = distributed_backend
173 def __init__(self, device, distributed_backend, bucket_size, model): argument
176 distributed_backend,
249 distributed_backend=args.distributed_backend,
259 distributed_backend=args.distributed_backend,
283 "distributed_backend": args.distributed_backend,
|
| H A D | README.md | 120 distributed_backend: nccl vs nccl
|