github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-04 02:26:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fx/tuning] tune performance on rotor with meta info. (#1599)

Browse Source
This commit is contained in:
Super Daniel
2022-09-15 14:46:36 +08:00
committed by GitHub
parent a7cda6f57d
commit cd5cf2bcc9
7 changed files with 96 additions and 107 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/fx/profiler/__init__.py
View File

@@ -7,4 +7,4 @@ else:
from .experimental import meta_profiler_function, meta_profiler_module, profile_function, profile_method, profile_module
from .dataflow import GraphInfo
from .memory import parameter_size, activation_size
from .memory import parameter_size, activation_size, is_inplace