ColossalAI/colossalai/fx
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
..
codegen [autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764) 2022-11-01 10:43:15 +08:00
passes [analyzer] a minimal implementation of static graph analyzer (#2852) 2023-03-10 13:21:05 +08:00
profiler [NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style (#3256) 2023-03-29 15:22:21 +08:00
tracer [fx] allow native ckpt trace and codegen. (#2438) 2023-01-11 13:49:59 +08:00
__init__.py [fx] metainfo_trace as an API. (#1873) 2022-11-10 20:58:37 +08:00
_compatibility.py [fx] meta registration compatibility (#3253) 2023-03-27 15:22:17 +08:00
_meta_regist_12.py [fx] meta registration compatibility (#3253) 2023-03-27 15:22:17 +08:00
_meta_regist_13.py [fx] meta registration compatibility (#3253) 2023-03-27 15:22:17 +08:00
graph_module.py [fx] allow control of ckpt_codegen init (#2498) 2023-01-18 17:02:46 +08:00
proxy.py [hotfix] fix coloproxy typos. (#1519) 2022-08-30 11:39:03 +08:00