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

..

__init__.py

[auto-parallel] add auto-offload feature (#3154)

2023-03-21 14:17:41 +08:00

amp_optimizer.py

[npu] change device to accelerator api (#5239)

2024-01-09 10:20:05 +08:00

base_offload_module.py

[hotfix] Fix examples no pad token & auto parallel codegen bug; (#5606)

2024-04-18 18:15:50 +08:00

mem_optimize.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

region_manager.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

region.py

[hotfix] Fix examples no pad token & auto parallel codegen bug; (#5606)

2024-04-18 18:15:50 +08:00

runtime.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

solver.py

[npu] change device to accelerator api (#5239)

2024-01-09 10:20:05 +08:00

training_simulator.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00

util.py

[misc] update pre-commit and run all files (#4752)

2023-09-19 14:20:26 +08:00