pytorch
e16979c9 - [threaded_pg] full rewrite of MultiThreadedTestCase to enable device_type tests (#91650)

Commit
1 year ago
[threaded_pg] full rewrite of MultiThreadedTestCase to enable device_type tests (#91650) This PR did a full rewrite of MultiThreadedTestCase, to make it more aligned with the MultiProcessTestCase, also changed how it do spawning and testing, so that we could embed thread local states when running tests. This PR enables device_type tests to work with MultiThreadedTestCase Pull Request resolved: https://github.com/pytorch/pytorch/pull/91650 Approved by: https://github.com/XilunWu
Author
Committer
Parents
Loading