6 Commits

Author	SHA1	Message	Date
Hanzhi Zhou	6192e9b8fe	[Core][Distributed] Refactor ipc buffer init in CustomAllreduce (#10030) ... Signed-off-by: Hanzhi Zhou <hanzhi713@gmail.com>	2024-11-06 23:50:47 -08:00
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