Method RequestStream

catboost/cuda/cuda_lib/cuda_manager.cpp:227–244

}

TComputationStream TCudaManager::RequestStream() {
    if (FreeStreams.size() == 0) {
        TDistributedObject<ui32> stream = CreateDistributedObject<ui32>();
        for (ui64 dev = 0; dev < stream.DeviceCount(); ++dev) {
            if (IsActiveDevice[dev]) {
                stream.Set(dev, GetState().Devices[dev]->RequestStream());
            } else {
                stream.Set(dev, 0);
            }
        }
        FreeStreams.push_back(Streams.size());
        Streams.push_back(stream);
    }

    ui32 id = FreeStreams.back();
    FreeStreams.pop_back();
    return TComputationStream(id, this);
}

bool TCudaManager::HasRemoteDevices() const {
    for (auto dev : State->Devices) {