Searched refs:UintToInt (Results 1 – 17 of 17) sorted by relevance
61 if (ret != UintToInt(PROF_SUCCESS)) { in PluginInit()122 if (cb_ret != UintToInt(PROF_SUCCESS)) { in StartupProfiling()166 if (cb_ret != UintToInt(PROF_SUCCESS)) { in ProfStartUp()227 if (ret != UintToInt(PROF_SUCCESS)) { in CallMsprofReport()
28 total_node_num_ = UintToInt(initial_total_node_num_); in InitNode()62 if (meta->rank_id() != UINT32_MAX && UintToInt(meta->rank_id()) <= next_server_rank_id_) { in NextRankId()101 if (meta->rank_id() != UINT32_MAX && UintToInt(meta->rank_id()) <= next_worker_rank_id_) { in NextRankId()270 next_server_rank_id_ = UintToInt(*min_rank_id - 1); in ResetMetadata()308 int32_t NodeManager::worker_num() const { return UintToInt(meta_data_->worker_num); } in worker_num()310 int32_t NodeManager::server_num() const { return UintToInt(meta_data_->server_num); } in server_num()
53 global_rank = UintToInt(rank_id); in GetRank()93 device_num = UintToInt(world_rank_size); in PipelineSplit()
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()
81 …pb_message->ParseFromArray(message_buffer_.get(), UintToInt(message_header_.message_meta_length_))… in ReceiveMessage()
86 mindspore::device::GpuBufferMgr::GetInstance().set_device_id(UintToInt(device_id)); in set_cur_device_id()
81 … CHECK_OP_RET_WITH_EXCEPT(CudaDriver::SetDevice(UintToInt(device_id_)), "Failed to set device id"); in Init()
180 device_id_ = UintToInt(device_id); in Load()
110 inline int32_t UintToInt(uint32_t u) { in UintToInt() function
243 bool ret = device::gpu::CudaDriver::SetDevice(UintToInt(device_id)); in Run()
995 auto ret = rtCtxCreate(&rt_context_, 0, UintToInt(device_id_)); in CreateContext()1010 ret = rtSetDevice(UintToInt(device_id_)); in InitDevice()1065 ret = rtDeviceReset(UintToInt(device_id)); in ResetDevice()
112 auto ret = rtSetDevice(UintToInt(device_id)); in InitDevice()
527 if (!CudaDriver::SetDevice(UintToInt(device_context_key_.device_id_))) { in BindDeviceToCurrentThread()
110 bool ret = device::gpu::CudaDriver::SetDevice(UintToInt(device_id)); in Init()
2901 device_num = UintToInt(world_rank_size); in GetCommInfo()2918 global_rank = UintToInt(rank_id); in GetCommInfo()
1600 return adpt->getOutput(Convert(node), UintToInt(index_stack.top())); in GetHandler()