ColossalAI/colossalai/fx/profiler
Super Daniel 112a1f0a8f
[hotfix] avoid conflict of meta registry with torch 1.13.0. (#1530)
* [hotfix] avoid conflict of meta registry with torch 1.13.0.

* [hotfix] avoid conflict of meta registry with torch 1.13.0.
2022-09-01 15:31:21 +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 [hotfix] avoid conflict of meta registry with torch 1.13.0. (#1530) 2022-09-01 15:31:21 +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