[torch.compile] register allreduce operations as custom ops (#8526)

Browse Source

...

This commit is contained in:

youkaichao

2024-09-16 22:57:57 -07:00

committed by GitHub

parent ee2bceaaa6

commit 99aa4eddaf

9 changed files with 137 additions and 50 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

0

tests/compile/__init__.py Normal file