Searched refs:original_device (Results 1 – 2 of 2) sorted by relevance
1199 int original_device = 0; in CreateDevices() local1201 cudaError_t err = cudaGetDevice(&original_device); in CreateDevices()1207 hipError_t err = hipGetDevice(&original_device); in CreateDevices()1234 "cudaDeviceGetStreamPriorityRange() on GPU:", original_device, in CreateDevices()1237 VLOG(1) << "Cuda stream priority range on GPU(" << original_device in CreateDevices()1259 "hipDeviceGetStreamPriorityRange() on GPU:", original_device, in CreateDevices()1262 VLOG(1) << "HIP stream priority range on GPU(" << original_device in CreateDevices()1271 err = cudaSetDevice(original_device); in CreateDevices()1273 return errors::Internal("cudaSetDevice() on GPU:", original_device, in CreateDevices()1277 err = hipSetDevice(original_device); in CreateDevices()[all …]
244 original_device = new_op.device254 with _MaybeDevice(original_device):