github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-31 19:35:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
72341e65f4
ColossalAI/tests/test_autochunk
History
oahzxl 72341e65f4
[auto-chunk] support extramsa (#3) (#2504)
2023-01-20 10:13:03 +08:00
..
benchmark_simple_evoformer.py [autochunk] support evoformer tracer (#2485) 2023-01-16 19:25:05 +08:00
test_evoformer_codegen.py [auto-chunk] support extramsa (#3) (#2504) 2023-01-20 10:13:03 +08:00
test_extramsa_codegen.py [auto-chunk] support extramsa (#3) (#2504) 2023-01-20 10:13:03 +08:00
test_simple_evoformer_codegen.py [auto-chunk] support extramsa (#3) (#2504) 2023-01-20 10:13:03 +08:00
test_simple_evoformer_search.py [auto-chunk] support extramsa (#3) (#2504) 2023-01-20 10:13:03 +08:00