Searched refs:SetDevice (Results 1 – 9 of 9) sorted by relevance
67 static bool SetDevice(int index);
27 …CHECK_OP_RET_WITH_EXCEPT(CudaDriver::SetDevice(SizeToInt(cur_dev_id_)), "Failed to set current dev… in InitDevice()
242 bool CudaDriver::SetDevice(int index) { in SetDevice() function in mindspore::device::gpu::CudaDriver
81 … CHECK_OP_RET_WITH_EXCEPT(CudaDriver::SetDevice(UintToInt(device_id_)), "Failed to set device id"); in Init()
61 bool ret = device::gpu::CudaDriver::SetDevice(UintToInt(device_id_)); in InitEnv()210 bool ret = device::gpu::CudaDriver::SetDevice(UintToInt(device_id_)); in Run()
30 bool ret = device::gpu::CudaDriver::SetDevice(UintToInt(device_id)); in InitDevice()
243 bool ret = device::gpu::CudaDriver::SetDevice(UintToInt(device_id)); in Run()
527 if (!CudaDriver::SetDevice(UintToInt(device_context_key_.device_id_))) { in BindDeviceToCurrentThread()
110 bool ret = device::gpu::CudaDriver::SetDevice(UintToInt(device_id)); in Init()