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> |
||
---|---|---|
.. | ||
passes | ||
tracer | ||
__init__.py | ||
codegen.py | ||
graph_module.py | ||
node_util.py | ||
symbolic_profile.py |