Fix race condition in ThreadPool::workOnTasksUntilCompleted (#14833)
Summary:
Resolves #14704
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14833
Differential Revision: D13405211
Pulled By: highker
fbshipit-source-id: 8552d51eeb5d3af0ed66c461e5ddfeb9ae2926bd
