Class DeviceSpVecDouble

XM/include/Utils/memory.h:1080–1139

sparse vector wrapper on device: double type

// sparse vector wrapper on device: double type
class DeviceSpVecDouble {
    public:
        int gpu_id;
        int size; 
        int nnz;
        int* indices;
        double* vals; 
        cusparseSpVecDescr_t cusparse_descr;

        DeviceSpVecDouble(): gpu_id(0), size(0), nnz(0), 
            indices(nullptr), vals(nullptr), cusparse_descr(NULL) {}
        DeviceSpVecDouble(const int gpu_id, const int size, const int nnz):
            gpu_id(gpu_id), size(size), nnz(nnz), 
            vals(nullptr), indices(nullptr), cusparse_descr(NULL) {
                this->allocate(gpu_id, size, nnz);
            }
        
        inline void allocate(const int gpu_id, const int size, const int nnz) {
            if (this->vals == nullptr) {
                this->gpu_id = gpu_id;
                this->size = size; 
                this->nnz = nnz;
                CHECK_CUDA( cudaSetDevice(this->gpu_id) );
                CHECK_CUDA( cudaMalloc((void**) &this->indices, sizeof(int) * this->nnz) );
                CHECK_CUDA( cudaMalloc((void**) &this->vals, sizeof(double) * this->nnz) );
                CHECK_CUSPARSE( cusparseCreateSpVec(
                    &this->cusparse_descr, this->size, this->nnz, 
                    this->indices, this->vals, 
                    CUSPARSE_INDEX_32I, CUSPARSE_INDEX_BASE_ZERO, CUDA_R_64F 
                ) );
            }
            return;
        }

        inline double get_norm(const DeviceBlasHandle& cublas_H) {
            double norm;
            CHECK_CUDA( cudaSetDevice(this->gpu_id) );
            CHECK_CUBLAS( cublasDnrm2_v2(
                cublas_H.cublas_handle, this->nnz, this->vals, 1, &norm
            ) );
            return norm;
        }

        ~DeviceSpVecDouble() {
            CHECK_CUDA( cudaSetDevice(this->gpu_id) );
            if (this->indices != nullptr) {
                CHECK_CUDA( cudaFree(this->indices) );
                this->indices = nullptr;
            }
            if (this->vals != nullptr) {
                CHECK_CUDA( cudaFree(this->vals) );
                this->vals = nullptr;
            }
            if (this->cusparse_descr != NULL) {
                CHECK_CUSPARSE( cusparseDestroySpVec(this->cusparse_descr) );
                this->cusparse_descr = NULL;
            }
            // std::cout << "DeviceSpVecDouble destructor called!" << std::endl;