ColossalAI/colossalai/fx
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
..
codegen [hotfix] Fix examples no pad token & auto parallel codegen bug; (#5606) 2024-04-18 18:15:50 +08:00
passes [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
profiler [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
tracer [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
__init__.py [fx] metainfo_trace as an API. (#1873) 2022-11-10 20:58:37 +08:00
_compatibility.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
_meta_regist_12.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
_meta_regist_13.py [fx] meta registration compatibility (#3253) 2023-03-27 15:22:17 +08:00
graph_module.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00
proxy.py [misc] update pre-commit and run all files (#4752) 2023-09-19 14:20:26 +08:00