youkaichao eea55cca5b [1/N] torch.compile user interface design (#10237) ... Signed-off-by: youkaichao <youkaichao@gmail.com>		2024-11-11 18:01:06 -08:00
..
piecewise	[1/N] torch.compile user interface design (#10237)	2024-11-11 18:01:06 -08:00
__init__.py	[torch.compile] register allreduce operations as custom ops (#8526)	2024-09-16 22:57:57 -07:00
backend.py	[torch.compile] Fuse RMSNorm with quant (#9138)	2024-11-08 21:20:08 +00:00
test_basic_correctness.py	[torch.compile] rework test plans (#9866)	2024-10-31 22:20:17 -07:00
test_full_graph.py	[torch.compile] rework compile control with piecewise cudagraph (#9715)	2024-10-29 23:03:49 -07:00
test_fusion.py	[torch.compile] Fuse RMSNorm with quant (#9138)	2024-11-08 21:20:08 +00:00
test_wrapper.py	[tpu][misc] fix typo (#8260)	2024-09-06 22:40:46 -07:00
utils.py	[Bugfix] Enable some fp8 and quantized fullgraph tests (#10171)	2024-11-09 08:01:27 +00:00