Searched refs:test_multi_device (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/test/inductor/ | ||
| D | test_foreach.py | 742 def test_multi_device(self): member in ForeachTests |
| D | test_aot_inductor.py | 275 def test_multi_device(self): member in AOTInductorTestsTemplate |
| D | test_torchinductor.py | 3542 def test_multi_device(self): member in CommonTemplate |