Commit de7ac60

authored and

committed

Add out= variants for cuda.comm.broadcast/gather/scatter (#39681)

Summary: Partially fixes #38911 Pull Request resolved: #39681 Differential Revision: D22161342 Pulled By: mrshenli fbshipit-source-id: 60295077159b02087823e93bb6ebac9d70adea0a

1 parent e664458 commit de7ac60Copy full SHA for de7ac60

6 files changed

test
- test_cuda.py
torch
- csrc/cuda
- cuda
  - _utils.py
  - comm.py

Comments

(0)

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Commit de7ac60

File tree

0 commit comments