vector.cpp

Go to the documentation of this file.

/// @file core/backends/omp/vector.cpp
/// @brief OpenMP backend  -- parallelised vector operations
 
#include "core/vector.hpp"
#include "../seq/impl.hpp"
 
namespace num::backends::omp {
 
void scale(Vector& v, real alpha) {
#ifdef NUMERICS_HAS_OMP
    const idx n = v.size();
#   pragma omp parallel for schedule(static)
    for (idx i = 0; i < n; ++i) v[i] *= alpha;
#else
    num::backends::seq::scale(v, alpha);
#endif
}
 
void axpy(real alpha, const Vector& x, Vector& y) {
#ifdef NUMERICS_HAS_OMP
    const idx n = x.size();
#   pragma omp parallel for schedule(static)
    for (idx i = 0; i < n; ++i) y[i] += alpha * x[i];
#else
    num::backends::seq::axpy(alpha, x, y);
#endif
}
 
real dot(const Vector& x, const Vector& y) {
#ifdef NUMERICS_HAS_OMP
    real sum = 0;
    const idx n = x.size();
#   pragma omp parallel for reduction(+:sum) schedule(static)
    for (idx i = 0; i < n; ++i) sum += x[i] * y[i];
    return sum;
#else
    return num::backends::seq::dot(x, y);
#endif
}
 
} // namespace num::backends::omp