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> |
||
---|---|---|
.. | ||
__init__.py | ||
activation_checkpoint_codegen.py |