mirror of https://github.com/hpcaitech/ColossalAI
d83c633ca6
* fix no pad token bug * fixed some auto parallel codegen bug, but might not run on torch 2.1 --------- Co-authored-by: Edenzzzz <wtan45@wisc.edu> |
||
---|---|---|
.. | ||
autochunk_codegen.py | ||
estimate_memory.py | ||
reorder_graph.py | ||
search_chunk.py | ||
select_chunk.py | ||
trace_flow.py | ||
trace_indice.py | ||
utils.py |