Searched refs:initial_cuda_device (Results 1 – 1 of 1) sorted by relevance
172 int initial_cuda_device; in CreateNcclClique() local173 XLA_CUDA_RETURN_IF_ERROR(cudaGetDevice(&initial_cuda_device)); in CreateNcclClique()175 [&] { XLA_CUDA_WARN_IF_ERROR(cudaSetDevice(initial_cuda_device)); }); in CreateNcclClique()