ColossalAI/colossalai/fx/profiler/experimental
dayellow 204ca2f09a [NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style (#3256)
Co-authored-by: Minghao Huang <huangminghao@luchentech.com>
2023-03-29 15:22:21 +08:00
..
profiler_function [doc] add deepspeed citation and copyright (#2996) 2023-03-04 20:08:11 +08:00
profiler_module [NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style (#3256) 2023-03-29 15:22:21 +08:00
__init__.py [fx] refactor memory utils and extend shard utils. (#1754) 2022-10-26 14:24:41 +08:00
constants.py [fx/meta/rpc] move _meta_registration.py to fx folder / register fx functions with compatibility checks / remove color debug (#1710) 2022-10-18 10:44:23 +08:00
profiler.py [fx] refactor memory utils and extend shard utils. (#1754) 2022-10-26 14:24:41 +08:00
registry.py
shard_utils.py [fx] refactor memory utils and extend shard utils. (#1754) 2022-10-26 14:24:41 +08:00