Resolve deadlock (#16805)

Commit

2 years ago

Resolve deadlock (#16805) Resolve deadlock --------- Co-authored-by: Randy Shuai <rashuai@microsoft.com>