[WIP][resubmit] Don't #define NUM_THREADS (#68008)
Summary:
This reverts commit 9e8016d8c48e9c99addad93112f99d3375a0fbc7.
cc jeffdaily sunway513 jithunnair-amd ROCmSupport KyleCZH
Pull Request resolved: https://github.com/pytorch/pytorch/pull/68008
Reviewed By: H-Huang
Differential Revision: D32254779
Pulled By: ngimel
fbshipit-source-id: 38ec415199f62a1e58000abe3e34ac91898a94ae
