Class TSingleMapping

catboost/cuda/cuda_lib/mapping.h:83–154

    };

    class TSingleMapping: public TMappingBase<TSingleMapping> {
    protected:
        ui64 Count = 0;
        ui32 DeviceId = 0;

    public:
        using TMappingBase::TMeta;

        explicit TSingleMapping(ui32 devId = 0, ui64 count = 0, ui64 size = 1)
            : TMappingBase(size)
            , Count(count)
            , DeviceId(devId)
        {
        }

        explicit TSingleMapping(TVector<TSlice>&& slices, ui64 singleObjectSize = 1)
            : TMappingBase(singleObjectSize)
        {
            CB_ENSURE(slices.size() == NCudaLib::GetCudaManager().GetDeviceCount());

            bool initDone = false;
            for (ui32 i = 0; i < slices.size(); ++i) {
                if (slices[i].Size()) {
                    CB_ENSURE(!initDone);
                    initDone = true;
                    DeviceId = i;
                    Count = slices[i].Size();
                }
            }
        }

        TSlice GetObjectsSlice() const {
            return {0, Count};
        }

        ui64 CountAt(ui64 dev) const {
            if (dev != DeviceId) {
                return 0;
            }
            return Count;
        }

        ui32 GetDeviceId() const {
            return DeviceId;
        }

        TSingleMapping ToLocalSlice(const TSlice& slice) const {
            CB_ENSURE(GetObjectsSlice().Contains(slice));
            return TSingleMapping(DeviceId, slice.Size(), SingleObjectSize());
        }

        TSlice DeviceSlice(ui64 dev) const {
            if (dev != DeviceId) {
                return {0, 0};
            }
            return {0, Count};
        }