Home
last modified time | relevance | path

Searched refs:test_device_synchronize (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_cuda.py62 def test_device_synchronize(): function
86 self.assertTrue(test_device_synchronize)
87 FileCheck().check("cuda::synchronize(").run(test_device_synchronize.graph)
/aosp_15_r20/external/pytorch/test/
H A Dtest_cuda_sanitizer.py331 def test_device_synchronize(self): member in TestEventHandler
H A Dtest_mps.py7989 def test_device_synchronize(self): member in TestMPS