PR #19757 Adapt memory optimizer to fit PHI2

Adapt memory optimizer to fit PHI2 #19757

pengwa merged 7 commits into main from pengwa/recompute_refinement

fix transformer layer detection

88a3bd89

use default reversed typo order to avoid leaf node recompute node add…

22499870

add shape check to stop earlier for recompute subgraph detection

4bd3ea7e

pengwa added training

pengwa requested a review from

askhade 1 year ago

pengwa requested a review from

frank-dong-ms 1 year ago

fix build

abe34e2c

increase the maximum node

39c62858

refine comment

08b06b74

Merge branch 'main' of https://github.com/microsoft/onnxruntime into …

f6a2a731

askhade commented on 2024-03-06

askhade approved these changes on 2024-03-06

pengwa merged d9bf8561 into main 1 year ago

pengwa deleted the pengwa/recompute_refinement branch 1 year ago

Reviewers

askhade

frank-dong-ms

Assignees

No one assigned

Labels

training

Milestone

No milestone