unit test with multiple omp threads (#14958)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14958
Test with multiple threads
Reviewed By: jianyuh
Differential Revision: D13394791
fbshipit-source-id: 931a6c3bda15ebc816807e537dd0841c383e7a6f
