Allow layer-wise recompute #18566
pengwa
merged
27 commits into
main
from
pengwa/add_aggresive_recompute
full recompute
7164c2f8
github-advanced-security
commented
on 2023-11-23
docs
7217ad38
fix
eed2916d
minor
afc96744
fix memory inspector
1df5dd76
move files
deedc44e
typo
15a06408
allow layer-wise recompute
4a881966
pengwa
changed the title
fixes
f81a3d6f
add layerwise recompute test case
d79d171b
missing test file
c5ba05a7
refine a bit
3334d4d1
fixes
bf89c047
minor
38fa66fd
fix
e8817c71
Gelu is followed with a Reshape for gradient ops, but its NodeIndex …
cb4eb7ba
fix
486da22c
fix
c4e72bbc
Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
f3369a8b
fix CIs
e342f59f
Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
7c3bbc37
enable tile recompute UT
30aebbce
askhade
commented
on 2023-12-05
askhade
commented
on 2023-12-05
refine review comments
b8e6088a
Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
0a5f4d45
zhijxu-MS
dismissed these changes
on 2023-12-08
lint
0fdfa4ba
typos
1d0c2bc1
typo
16020600
zhijxu-MS
approved these changes
on 2023-12-11
askhade
approved these changes
on 2023-12-11
Reviewers
Assignees
No one assigned
Labels
training
Milestone
No milestone
Login to write a write a comment.
Login via GitHub