tree 96e3eb27ff00ef90d231b43a7dfada942f17fb65
parent e89b150a39b3bc9ebaeaf52b9e9628405b09f662
author Akshit Khurana <axit@fb.com> 1622134125 -0700
committer Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> 1622134192 -0700

Fix pthreadpool guard test (#58977)

Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58977

* Test was flaky as part of it ran async
* Remove async part to test only the functionality added

Test Plan:
regular test:

`buck test mode/dev //caffe2/aten:test_thread_pool_guard -- --exact 'caffe2/aten:test_thread_pool_guard - TestThreadPoolGuard.TestRunWithGuard' --run-disabled`

stress test:

`buck test mode/dev //caffe2/aten:test_thread_pool_guard -- --exact 'caffe2/aten:test_thread_pool_guard - TestThreadPoolGuard.TestRunWithGuard' --run-disabled --jobs 18 --stress-runs 10 --record-results`

Reviewed By: kimishpatel

Differential Revision: D28703064

fbshipit-source-id: be19da3f42f44288afc726bdb2f40342eee26e01