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