youkaichao 99aa4eddaf

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

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

..

__init__.py

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

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

test_full_graph.py

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

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

test_wrapper.py

[tpu][misc] fix typo (#8260)

2024-09-06 22:40:46 -07:00