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 |
||
|---|---|---|
| .. | ||
| benchmark_simple_evoformer.py | ||
| test_evoformer_codegen.py | ||
| test_simple_evoformer_codegen.py | ||
| test_simple_evoformer_search.py | ||