Searched refs:old_cuda_device (Results 1 – 1 of 1) sorted by relevance
888 int old_cuda_device = 0; in ConvertAfterShapes() local889 auto err = cudaGetDevice(&old_cuda_device); in ConvertAfterShapes()893 VLOG(1) << "Current cuda device is " << old_cuda_device; in ConvertAfterShapes()948 cudaSetDevice(old_cuda_device); in ConvertAfterShapes()