[DTensor][fix] MultiThreadedTestCase misses _tls object and it won't reflect in CI (#93832)

XilunWu · pytorchmergebot · commit 966030f7c75b · 2023-02-02T07:56:44.000Z
Pull Request resolved: #93832 Approved by: https://github.com/wanchaol
diff --git a/torch/testing/_internal/common_distributed.py b/torch/testing/_internal/common_distributed.py
@@ -1025,6 +1025,7 @@ def _run(cls, test_name, rank, world_size):
         # every thread have the same value. This would be relevant when we use op db tests, where it
         # needs those states to be set i.e. using instantiate_device_type_tests()
         # TODO: figure out a better way to do this
+        self._tls = threading.local()
         self._tls.precision = TestCase._precision
         self._tls.rel_tol = TestCase._rel_tol
 
