[Remat][Bugfix] Enhance the rematerialization pass to handle nested tuples#17

Merged

comaniac merged 5 commits intoawslabs:mainfrom zhouyuan1119:remat_fix

Apr 17, 2022

+110-16

Commits on Apr 13, 2022

[Remat][Bugfix] Fix TensorAnalyzer to handle nested tuples.
zhouyuan1119
committed

Commits on Apr 14, 2022

Commits on Apr 15, 2022

[Remat][Bugfix] Add test for nested tuples.
zhouyuan1119
committed

Commits on Apr 16, 2022

[Remat][Bugfix] Remove log printing.
zhouyuan1119
committed