You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/fx/profiler
YuliangLiu0306 fbd2a9e05b
[hotfix] meta_tensor_compatibility_with_torch2
2 years ago
..
experimental [NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style (#3256) 2 years ago
__init__.py
constants.py
dataflow.py
memory_utils.py
opcount.py [hotfix] meta_tensor_compatibility_with_torch2 2 years ago
profiler.py
shard_utils.py
tensor.py