oahzxl 4953b4ace1 [autochunk] support evoformer tracer (#2485) ...

support full evoformer tracer, which is a main module of alphafold. previously we just support a simplifed version of it.
1. support some evoformer's op in fx
2. support evoformer test
3. add repos for test code

2023-01-16 19:25:05 +08:00

..

codegen

[autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764)

2022-11-01 10:43:15 +08:00

passes

[Pipeline Middleware ] Fix deadlock when num_microbatch=num_stage (#2156)

2022-12-23 11:38:43 +08:00

profiler

[autochunk] support evoformer tracer (#2485)

2023-01-16 19:25:05 +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

register meta func for rnn (#2159)

2022-12-21 23:06:18 +08:00

graph_module.py

[fx] allow native ckpt trace and codegen. (#2438)

2023-01-11 13:49:59 +08:00

proxy.py

[hotfix] fix coloproxy typos. (#1519)

2022-08-30 11:39:03 +08:00