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