Super Daniel 5cc849f6ce [fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515) ... * [fx] hack __torch_dispatch__ for meta tensor and autograd. * [fx] hack __torch_dispatch__ for meta tensor and autograd. * [fx] hack __torch_dispatch__ for meta tensor and autograd. * [fx] hack __torch_dispatch__ for meta tensor and autograd. * [fx] hack __torch_dispatch__ for meta tensor and autograd. * [fx] add bad case detections. * [fx] add bad case detections. * [fx] rename MetaTensor attributes. * [fx] fix unexpected error. * [fx] fix unexpected error. * [fx] fix unexpected error. * [fx] fix unexpected error. * [fx] fix unexpected error. * [fx] add register backward for native_batch_norm_backward. * [fx] add more meta backend support for nn.Modules. * [fx] add meta backend to support timm and torchvision models. * [fx] add meta hardswish for timm models.		2022-08-31 16:30:16 +08:00
..
codegen	[fx] Fix activation codegen dealing with checkpointing first op (#1510)	2022-08-27 19:39:21 +08:00
passes	[fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515)	2022-08-31 16:30:16 +08:00
profiler	[fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515)	2022-08-31 16:30:16 +08:00
tracer	[autoparallel] add cost graph class (#1481)	2022-08-25 17:19:59 +08:00
__init__.py	[fx] Add activation checkpoint solver rotor (#1496)	2022-08-26 10:34:21 +08:00
graph_module.py	[fx] Fix ckpt functions' definitions in forward (#1476)	2022-08-22 16:59:54 +08:00
proxy.py	[hotfix] fix coloproxy typos. (#1519)	2022-08-30 11:39:03 +08:00