204 Commits

Author	SHA1	Message	Date
youkaichao	7de49aa86c	[torch.compile] hide slicing under custom op for inductor (#8384)	2024-09-12 00:11:55 -07:00
youkaichao	ce2702a923	[tpu][misc] fix typo (#8260)	2024-09-06 22:40:46 -07:00
youkaichao	ce6bf3a2cf	[torch.compile] avoid Dynamo guard evaluation overhead (#7898) ... Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>	2024-08-28 16:10:12 -07:00
youkaichao	54bd9a03c4	register custom op for flash attn and use from torch.ops (#7536)	2024-08-15 22:38:56 -07:00