Я пытаюсь определить собственное сокращение для векторов сложных
Но размер моих векторов не фиксируется во время компиляции, поэтому я не уверен, как определить инициализатор для вектора в прагме declare reduction. То есть я просто не могу
initializer( omp_priv=TComplexVector(10,0) )
Но для векторов инициализатор нужен.
Как я могу передать предложению инициализатора размер вектора, который мне нужен во время выполнения? То, что у меня есть, ниже:
typedef std::vector<complex<float>> TCmplxVec;
void ComplexAdd(TCmplxVec & x,TCmplxVec & y){
for (int i=0;i<x.size();i++)
{
x.real()+= y.real();
//... same for imaginary part and other operations
}
}
#pragma omp declare reduction(AddCmplx: TCmplxVec: \
ComplexAdd(&omp_out, &omp_in)) initializer( \
omp_priv={TCmplxVec(**here I want a variable length**,0} )
void DoSomeOperation ()
{
//TCmplxVec vec is empty and anotherVec not
//so each thread runs the inner loop serially
#pragma omp parallel for reduction(AddCmplx: vec)
for ( n=0 ; n<10 ; ++n )
{
for (m=0; m<=someLength; ++m){
vec[m] += anotherVec[m+someOffset dependend on n and else];
}
}
}
1 ответ
Вам нужно немного покопаться, чтобы найти его в Интернете прямо сейчас, но в разделе 2.15 документа OpenMP Standard, где обсуждаются объявленные пользователем сокращения, вы обнаружите, что «специальный идентификатор omp_orig также может появиться в предложении инициализатора, и он будет относиться к хранилищу исходной переменной, которая будет уменьшена».
Таким образом, вы можете использовать initializer (omp_priv=TCmplxVec(omp_orig.size(),0)) или просто initalizer ( omp_priv(omp_orig) ) для инициализации вектора при редукции.
Итак, следующие работы (обратите внимание, что вам не нужно писать свою собственную процедуру; вы можете использовать std :: transform и std :: plus для добавления ваших векторов; вы также можете использовать std :: valarray, а не векторы, в зависимости от того, как вы используете их, для которых уже определен оператор +):
#include <complex>
#include <vector>
#include <algorithm>
#include <functional>
#include <iostream>
#include <omp.h>
typedef std::vector< std::complex<float> > TCmplxVec;
#pragma omp declare reduction( + : TCmplxVec : \
std::transform(omp_in.begin( ), omp_in.end( ), \
omp_out.begin( ), omp_out.begin( ), \
std::plus< std::complex<float> >( )) ) \
initializer (omp_priv(omp_orig))
int main(int argc, char *argv[]) {
int size;
if (argc < 2)
size = 10;
else
size = atoi(argv[1]);
TCmplxVec result(size,0);
#pragma omp parallel reduction( + : result )
{
int tid=omp_get_thread_num();
for (int i=0; i<std::min(tid+1,size); i++)
result[i] += tid;
}
for (int i=0; i<size; i++)
std::cout << i << "\t" << result[i] << std::endl;
return 0;
}
Запуск это дает
$ OMP_NUM_THREADS=1 ./reduction 8
0 (0,0)
1 (0,0)
2 (0,0)
3 (0,0)
4 (0,0)
5 (0,0)
6 (0,0)
7 (0,0)
$ OMP_NUM_THREADS=4 ./reduction 8
0 (6,0)
1 (6,0)
2 (5,0)
3 (3,0)
4 (0,0)
5 (0,0)
6 (0,0)
7 (0,0)
$ OMP_NUM_THREADS=8 ./reduction 8
0 (28,0)
1 (28,0)
2 (27,0)
3 (25,0)
4 (22,0)
5 (18,0)
6 (13,0)
7 (7,0)
Похожие вопросы
Связанные вопросы
Новые вопросы
c++
C++ — это язык программирования общего назначения. Изначально он разрабатывался как расширение C и имел аналогичный синтаксис, но теперь это совершенно другой язык. Используйте этот тег для вопросов о коде, который будет скомпилирован с помощью компилятора C++. Используйте тег версии для вопросов, связанных с конкретной стандартной версией [C++11], [C++14], [C++17], [C++20] или [C++23]. и т.д.
int S_private[10] = {0};, сделайтеint *S_private = new int[n](), а затем после критического участка сделайтеdelete[] S_private.