Class DeviceDnVecLongInt

XM/include/Utils/memory.h:1048–1077

dense vector wrapper on device: long int type

// dense vector wrapper on device: long int type
class DeviceDnVecLongInt {
    public:
        int gpu_id;
        int size;
        size_t* vals;

        DeviceDnVecLongInt(): gpu_id(0), size(0), vals(nullptr) {}
        DeviceDnVecLongInt(const int gpu_id, const int size): gpu_id(gpu_id), size(size), vals(nullptr) {
            this->allocate(this->gpu_id, this->size);
        }

        inline void allocate(const int gpu_id, const int size) {
            if (this->vals == nullptr) {
                this->gpu_id = gpu_id;
                this->size = size;
                CHECK_CUDA( cudaSetDevice(this->gpu_id) );
                CHECK_CUDA( cudaMalloc((void**) &this->vals, sizeof(size_t) * this->size) );
            }
            return;
        }

        ~DeviceDnVecLongInt() {
            CHECK_CUDA( cudaSetDevice(this->gpu_id) );
            if (this->vals != nullptr) {
                CHECK_CUDA( cudaFree(this->vals) );
                this->vals = nullptr;
            }
            // std::cout << "DeviceDnVecLongInt destructor called!" << std::endl;
        }
};

// sparse vector wrapper on device: double type
class DeviceSpVecDouble {