ColossalAI/colossalai/autochunk
Edenzzzz d83c633ca6
[hotfix] Fix examples no pad token & auto parallel codegen bug; (#5606)
* 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>
2024-04-18 18:15:50 +08:00
..
autochunk_codegen.py [hotfix] Fix examples no pad token & auto parallel codegen bug; (#5606) 2024-04-18 18:15:50 +08:00
estimate_memory.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
reorder_graph.py [autochunk] support multi outputs chunk search (#2538) 2023-02-01 13:18:51 +08:00
search_chunk.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
select_chunk.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
trace_flow.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
trace_indice.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
utils.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00