Home
last modified time | relevance | path

Searched defs:deviceGuard (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/c10d/
H A DProcessGroupNCCLTest.cpp88 at::cuda::OptionalCUDAGuard deviceGuard; in NCCLTest() local
170 at::cuda::OptionalCUDAGuard deviceGuard; in launchDeviceSleep() local
179 at::cuda::OptionalCUDAGuard deviceGuard; in valueInitialization() local
200 at::cuda::OptionalCUDAGuard deviceGuard; in valueInitializationForSparse() local
360 at::cuda::OptionalCUDAGuard deviceGuard; in run() local
376 at::cuda::OptionalCUDAGuard deviceGuard; in ReduceScatterBaseNCCLTest() local
H A DProcessGroupGlooAsyncTest.cpp88 at::cuda::OptionalCUDAGuard deviceGuard; in AsyncInputIsOutputTest() local
137 at::cuda::OptionalCUDAGuard deviceGuard; in run() local
163 at::cuda::OptionalCUDAGuard deviceGuard; in run() local
H A DProcessGroupGlooTest.cpp371 at::OptionalDeviceGuard deviceGuard; in testBroadcast() local