concurrency fixes (#58961)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/58446
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58961
Reviewed By: anjali411
Differential Revision: D28700307
Pulled By: Krovatkin
fbshipit-source-id: 1fed90c64e88aaf2824c48e006f66a9266d1e163
