onnxruntime
Avoid cudaStreamSync at the end of Forward/Backward
#9470

Merged

Avoid cudaStreamSync at the end of Forward/Backward #9470

SherlockNoMad merged 2 commits into master from bahuang/no_stream_sync

Skip cudaStreamSynchronize at the end of fw

23408f29

skip sync stream for end of backward

d7912804

SherlockNoMad added training

SherlockNoMad added component:ortmodule

SherlockNoMad requested a review from

pengwa 4 years ago

SherlockNoMad requested a review from

Lafi7e 4 years ago

SherlockNoMad requested a review from

weixingzhang 4 years ago

weixingzhang approved these changes on 2021-10-21

SherlockNoMad merged ff23b9ff into master 4 years ago

SherlockNoMad deleted the bahuang/no_stream_sync branch 4 years ago

Reviewers

weixingzhang

pengwa

Lafi7e

Assignees

No one assigned

Labels

training

Milestone

No milestone