github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-02 08:37:36 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
73afb6359489295eafa95378341b9da866fe6097
ColossalAI/colossalai/autochunk
History
Xuanlei Zhao 10c61de2f7 [autochunk] support vit (#3084) 
support vit for autochunk
* support some new ops for vit
* fix some bugs
* add test for vit
2023-03-10 10:23:26 +08:00
..
autochunk_codegen.py
[autochunk] support vit (#3084)
2023-03-10 10:23:26 +08:00
estimate_memory.py
[autochunk] refactor chunk memory estimation (#2762)
2023-03-08 16:22:30 +08:00
reorder_graph.py
[autochunk] support multi outputs chunk search (#2538)
2023-02-01 13:18:51 +08:00
search_chunk.py
[autochunk] refactor chunk memory estimation (#2762)
2023-03-08 16:22:30 +08:00
select_chunk.py
[autochunk] refactor chunk memory estimation (#2762)
2023-03-08 16:22:30 +08:00
trace_flow.py
[autochunk] support diffusion for autochunk (#2621)
2023-02-07 16:32:45 +08:00
trace_indice.py
[autochunk] support vit (#3084)
2023-03-10 10:23:26 +08:00
utils.py
[autochunk] support vit (#3084)
2023-03-10 10:23:26 +08:00