Class DeviceDnVecInt

XM/include/Utils/memory.h:1016–1045

dense vector wrapper on device: int type

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

        DeviceDnVecInt(): gpu_id(0), size(0), vals(nullptr) {}
        DeviceDnVecInt(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(int) * this->size) );
            }
            return;
        }

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

// dense vector wrapper on device: long int type
class DeviceDnVecLongInt {