xla
Let grad ckpt apply opt-barrier to all params and buffers
#7206

Merged

Let grad ckpt apply opt-barrier to all params and buffers #7206

alanwaketan merged 5 commits into master from alanwaketan/grad_ckpt

alanwaketan requested a review from

JackCaoG 2 years ago

alanwaketan assigned

alanwaketan 2 years ago

JackCaoG commented on 2024-06-05

JackCaoG approved these changes on 2024-06-05

initial commit

f094f5c0

Fix linters

a9cc304a

Fix test

acaa4df8

Improve test

b287944b

alanwaketan added tpuci

alanwaketan force pushed from 1cfb4695 to b287944b 2 years ago

fix ci

050bfec0

JackCaoG commented on 2024-06-06

alanwaketan merged aec27305 into master 2 years ago

alanwaketan deleted the alanwaketan/grad_ckpt branch 2 years ago

Reviewers

JackCaoG

Assignees

alanwaketan

Labels

None yet

Milestone

No milestone