mirror of https://github.com/hpcaitech/ColossalAI
4953b4ace1
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 |
||
|---|---|---|
| .. | ||
| autochunk_codegen.py | ||
| estimate_memory.py | ||
| reorder_graph.py | ||
| search_chunk.py | ||
| select_chunk.py | ||
| trace_flow.py | ||
| trace_indice.py | ||
| utils.py | ||