5 Commits

Author	SHA1	Message	Date
youkaichao	99aa4eddaf	[torch.compile] register allreduce operations as custom ops (#8526)	2024-09-16 22:57:57 -07:00
bnellnm	5467ac3196	[Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047)	2024-06-09 16:23:30 -04:00
Michael Goin	5f6d10c14c	[CI/Build] Enforce style for C++ and CUDA code with clang-format (#4722)	2024-05-22 07:18:41 +00:00
Hanzhi Zhou	f721096d48	[BugFix] Some fixes for custom allreduce kernels (#2760)	2024-03-21 23:02:58 -07:00
Hanzhi Zhou	380170038e	Implement custom all reduce kernels (#2192)	2024-01-27 12:46:35 -08:00