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
youkaichao	47790f3e32	[torch.compile] add a flag to disable custom op (#8488)	2024-09-14 13:07:16 -07:00
youkaichao	a36e070dad	[torch.compile] fix functionalization (#8480)	2024-09-14 09:46:04 -07:00
youkaichao	7de49aa86c	[torch.compile] hide slicing under custom op for inductor (#8384)	2024-09-12 00:11:55 -07:00
youkaichao	54bd9a03c4	register custom op for flash attn and use from torch.ops (#7536)	2024-08-15 22:38:56 -07:00