Форум программистов [Powered by Invision Power Board]

Версия для печати темы
Нажмите сюда для просмотра этой темы в оригинальном формате

Форум программистов > C/C++: Общие вопросы > all_gather для векторов

Автор: Хоббит 26.1.2011, 20:46

Использую boost::mpi для написания распределенной программы, решения системы дифференциальных уравнений.
Допустим порядок системы 3. А запускаем программу на 2х узлах. Тогда на 1 узле считаются 1 и 3 уравнения, а на 2 узле считается 2 уравнение.

Необходимо в одном месте собрать все значения в один вектор. Есть метод all_gather, но он собирает скалярные значения в вектор.
В моем же случае нужно собрать вектора в вектор. (при этом добавить значения всех векторов в один вектор, а не получить вектор векторов)

Допустим есть функция
vector<int> getResult();
Которая возвращает вектор результатов для уравнений, которые считает. (На 1 узле это вектор из 2х значений для 1 и 3 уравнения системы. На 2 узле это 1 значение.) Необходимо получить один вектор с всеми тремя значениями.

Есть ли для этого стандартная функция?

Пока сделал свою функцию.

Код


template<typename T> 
void my_all_gather(const communicator & comm, std::vector< T > & in_value, std::vector< T > & out_values)
{
    //Объявляем вектор векторов для сбора результатов стандартным способом
    std::vector< std::vector< T > > out_values_vector;
    // Стандартный способ
    all_gather(comm, in_value, out_values_vector);

    // Теперь переписываем вектор векторов в обычный вектор
    for (int i = 0; i < out_values_vector.size(); i++)
    {
        for (int j = 0; j < out_values_vector[i].size(); j++)
        {
            out_values->push_back(out_values_vector[i][j]);
        }
    }
}