Блог

Я собираюсь вычислить размер вектора. Если я напишу этот код:

     Eigen::Matrix<float, 6, 1, Eigen::ColMajor> staticVector;
    std::cout << "sizeInBytes: " << sizeof (staticVector) << std::endl;
 

вывод:

 sizeInBytes: 24
 

Но если я напишу:

     Eigen::Matrix<float, Eigen::Dynamic, 1, Eigen::ColMajor> dynamicVector;
    dynamicVector.resize(6);
    std::cout << "sizeInBytes: " << sizeof (dynamicVector) << std::endl;
 

вывод:

 sizeInBytes: 16
 

Хорошо, sizeof (vector) не показывает реальный размер динамического вектора. Что я должен использовать? Я бы попробовал sizeof (float) * vector.size() или sizeof (float) * vector.size() sizeof (vector), но не уверен, что это правильный вариант. Итак, что мне нужно?

Если Eigen::Matrix весь массив находится в стеке, то sizeof он должен возвращать правильное значение. Это было бы здорово для небольших матриц, но может стать опасным для самых больших, поскольку это может вызвать переполнение стека.

Если данные хранятся в куче, вам нужно создать метод, который добавляет память стека и кучи и возвращает ее. Это было бы возможным примером:

 template<class T, size_t ROWS, size_t COLS, Eigen::SomeEnumType>
class Matrix {
    // ...
  private:
    std::array<std::array<T, COLS>, ROWS> *data;

  public:
    Matrix() { data = new std::array<std::array<T, COLS>, ROWS>(); }

    // Assumes all dynamically allocated memory is the data.
    size_t sizeInBytes() const { return sizeof(T) * ROWS * COLS   sizeof(*this); }
}; 
 

О проблеме в комментарии:

Я пытаюсь передать динамический вектор на графический процессор с помощью Cuda

Я очень мало работал с CUDA, поэтому могу ошибаться, но IIRC — единственный способ передать данные в GPU с memcpy помощью операции, подобной операции. Это было бы очень сложно или невозможно для векторов, которые содержат данные стека и кучи далеко друг от друга. Лучшим способом было бы выделить все данные внутри объекта (чтобы они были непрерывными в памяти), а затем, если вы боитесь переполнения стека, просто выделите весь объект в куче:

 template<class T, size_t ROWS, size_t COLS, Eigen::SomeEnumType>
class Matrix {
    // ...

    std::array<std::array<T, COLS>, ROWS> data;
}; 

int main() {
  auto *mat = new Eigen::Matrix<float, 6, 1, Eigen::ColMajor>(...);
  void *GPU_mat = malloc_GPU(sizeof(*mat));
  memcpy_to_GPU(GPU_mat, mat, sizeof(*mat));
}