Xuanlei Zhao 10c61de2f7 [autochunk] support vit (#3084) ... support vit for autochunk * support some new ops for vit * fix some bugs * add test for vit		2023-03-10 10:23:26 +08:00
..
codegen	[autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764)	2022-11-01 10:43:15 +08:00
passes	[pipeline] Add Simplified Alpa DP Partition (#2507)	2023-03-07 10:34:31 +08:00
profiler	[autochunk] support vit (#3084)	2023-03-10 10:23:26 +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/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
_meta_registrations.py	support unet metainfo prop (#2544)	2023-02-02 16:19:26 +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