xla
[FSDP] Add profiler to ImageNet
#4754
Merged

[FSDP] Add profiler to ImageNet #4754

alanwaketan merged 3 commits into master from alanwaketan/profiler
alanwaketan
alanwaketan [FSDP] Add profiler to ImageNet
5d1276ef
alanwaketan alanwaketan requested a review from lsy323 lsy323 2 years ago
alanwaketan alanwaketan requested a review from JackCaoG JackCaoG 2 years ago
alanwaketan alanwaketan assigned alanwaketan alanwaketan 2 years ago
alanwaketan Fix linters
0757e7d5
JackCaoG
JackCaoG commented on 2023-03-10
lsy323
lsy323 approved these changes on 2023-03-13
lsy323
alanwaketan Move add_step_closure out from build_graph
cc3d82dc
alanwaketan
alanwaketan alanwaketan merged d8125517 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
Milestone