Class DeviceDnTen

XM/include/Utils/memory.h:282–502

// dense vector wrapper on device
template <typename T>
class DeviceDnTen {
public:
    size_s gpu_id = 0;
    size_s* dimensions = nullptr; // Array to store the size of each dimension
    size_s num_dims = 0;    // Number of dimensions
    size_l total_size = 1;
    T* vals = nullptr;           // Pointer to the array of values
    T* vals_host = nullptr;
    cusparseDnVecDescr_t cusparse_descr = NULL;
    cusparseDnMatDescr_t cusparse_descr_mat = NULL;
    // bool* isdestroied = new bool(false);

    DeviceDnTen(){}

    DeviceDnTen(const size_s num_dims, const size_s* dimensions) 
        : num_dims(num_dims){
        this->dimensions = (size_s*)malloc(sizeof(size_s) * num_dims);
        std::memcpy(this->dimensions, dimensions, sizeof(size_s) * num_dims); // Copy dimensions
        
        // Calculate total size
        total_size = 1;
        for (size_s i = 0; i < num_dims; ++i) {
            total_size *= dimensions[i];
        }

        this->allocate();
    }

    DeviceDnTen(const size_s gpu_id, const size_s num_dims, const size_s* dimensions) 
        : gpu_id(gpu_id), num_dims(num_dims) {
        this->dimensions = (size_s*)malloc(sizeof(size_s) * num_dims);
        std::memcpy(this->dimensions, dimensions, sizeof(size_s) * num_dims); // Copy dimensions
        
        // Calculate total size
        total_size = 1;
        for (size_s i = 0; i < num_dims; ++i) {
            total_size *= dimensions[i];
        }

        this->allocate();
    }

    DeviceDnTen(std::initializer_list<size_s> dims) : num_dims(dims.size()) {
        dimensions = new size_s[num_dims];
        std::copy(dims.begin(), dims.end(), dimensions);

        // Calculate total number of elements
        total_size = 1;
        for (size_s i = 0; i < num_dims; ++i) {
            total_size *= dimensions[i]; 
        }
        this->allocate();
    }

    DeviceDnTen(const DeviceDnTen<T>& other) 
        : gpu_id(other.gpu_id), num_dims(other.num_dims), total_size(other.total_size) {

        this->dimensions = (size_s*)malloc(sizeof(size_s) * num_dims);