Function ReadToDistributedObjectVec

catboost/cuda/cuda_lib/cuda_buffer.h:672–692

    template <class T>
    inline TVector<TDistributedObject<std::remove_const_t<T>>> ReadToDistributedObjectVec(const TStripeBuffer<T>& src) {
        using T_ = std::remove_const_t<T>;
        TVector<T_> tmp;
        src.Read(tmp);
        ui32 devCount = NCudaLib::GetCudaManager().GetDeviceCount();
        TVector<TDistributedObject<T_>> result;

        for (ui32 dev = 0; dev < devCount; ++dev) {
            CB_ENSURE(src.GetMapping().DeviceSlice(dev).Size() == src.GetMapping().DeviceSlice(0).Size());
        }
        ui32 size = static_cast<ui32>(tmp.size() / devCount);

        for (ui32 i = 0; i < size; ++i) {
            TDistributedObject<T_> obj = NCudaLib::GetCudaManager().CreateDistributedObject<T_>();
            for (ui32 dev = 0; dev < devCount; ++dev) {
                obj.Set(dev, tmp[i + dev * size]);
            }
            result.push_back(obj);
        }
        return result;
    }
}