Searched refs:global_device_rank (Results 1 – 1 of 1) sorted by relevance
144 Status DeviceManager::Init(const RankList &devices, int64_t global_device_rank, const RankList &sta… in Init() argument180 std::shared_ptr<Device> dev = std::make_shared<Device>(global_device_rank); in Init()183 global_rank_ = global_device_rank; in Init()185 stage_id_ = global_device_rank / static_cast<const int64_t>(devices.size() / stage_map.size()); in Init()198 MS_LOG(INFO) << "The device num: " << devices.size() << ", rank id: " << global_device_rank in Init()