Super Daniel e7383f578b [fx] add rules to linearize computation graphs for searching. (#1461) ... * [fx] polish ckpt_test. * [fx] add rules to linearize computation graphs for searching. * [fx] remove chen_sqrt for sake of simplicity * [fx] fix inconsistencies.		2022-08-17 14:47:12 +08:00
..
codegen	[fx] Add use_reentrant=False to checkpoint in codegen (#1463)	2022-08-17 10:34:50 +08:00
passes	[fx] add rules to linearize computation graphs for searching. (#1461)	2022-08-17 14:47:12 +08:00
tracer	[fx] patched torch.max and data movement operator (#1391)	2022-08-01 15:31:50 +08:00
__init__.py	[fx] supported data-dependent control flow in model tracing (#1185)	2022-06-29 15:05:25 +08:00
proxy.py	[doc] update rst and docstring (#1351)	2022-07-21 15:54:53 +08:00