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

..

autochunk_codegen.py

[autochunk] support evoformer tracer (#2485)

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

estimate_memory.py

adapt new fx

2023-01-10 11:56:00 +08:00

reorder_graph.py

add doc

2023-01-10 09:59:47 +08:00

search_chunk.py

update doc

2023-01-10 10:44:07 +08:00

select_chunk.py

rename trace_index to trace_indice

2023-01-09 17:25:13 +08:00

trace_flow.py

[autochunk] support evoformer tracer (#2485)

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

trace_indice.py

[autochunk] support evoformer tracer (#2485)

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

utils.py

[autochunk] support evoformer tracer (#2485)

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