ColossalAI/colossalai/fx/profiler
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
..
profiler_function [fx] add more op patches for profiler and error message for unsupported ops. (#1495) 2022-08-25 23:11:13 +08:00
profiler_module [fx] add more op patches for profiler and error message for unsupported ops. (#1495) 2022-08-25 23:11:13 +08:00
__init__.py [fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515) 2022-08-31 16:30:16 +08:00
_meta_registrations.py [fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515) 2022-08-31 16:30:16 +08:00
meta_tensor.py [fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515) 2022-08-31 16:30:16 +08:00
profiler.py [fx] hack __torch_dispatch__ for meta tensor and autograd. (#1515) 2022-08-31 16:30:16 +08:00
registry.py [fx] add profiler for fx nodes. (#1480) 2022-08-24 16:22:44 +08:00