RcppParallel中二次型的并行计算

时间:2016-11-01 21:31:14

标签: c++ r parallel-processing rcpp

我想编写一个代码来使用RcppParallel计算v ^ T Av。这里v是大小为n的向量,A是n乘n矩阵。我的想法是以平行方式评估Av,然后使用v计算此向量的内积。这是我的代码:

#include <Rcpp.h>
// [[Rcpp::depends(RcppParallel)]]
#include <RcppParallel.h>
#include<iostream>
#include<vector>
#include<algorithm>
#include<numeric>

using namespace std;
using namespace Rcpp;
using namespace RcppParallel;

struct Par_MatVec_Mult: public Worker {

  const RMatrix<double> Mat;
  const vector<double> Vec;
  vector<double> output;

  Par_MatVec_Mult(RMatrix<double> A, vector<double> v, vector<double> Av): \ 
  vector<double> Av): Mat(A), Vec(v), output(Av) { }
  void operator() ( size_t begin, size_t end) {

    for( size_t i = begin; i < end; i++ ) {
      RMatrix<double>::Row rowi = Mat.row(i);
       output.at(i) = inner_product( rowi.begin(), rowi.end(),Vec.begin(), \
       0.0 );
    }
  }
};

// [[Rcpp::export]]
double Parallel_MatVec_Mult( NumericMatrix A, vector<double> v ){

  vector<double> b( A.nrow(), 0.0 );
  Par_MatVec_Mult Av(A, v, b);
  parallelFor(0, A.nrow(), Av);
  return inner_product( Av.output.begin(), Av.output.end(), v.begin(), 0.0   );
}

编译时我收到以下错误。

no known conversion for argument 1 from 'Rcpp::NumericMatrix {akaRcpp::Matrix<14>}' 
to 'RcppParallel::RMatrix<double>'

我很清楚,NumericMatrix和RMatrix是两个不同的对象。但是,我不知道它们之间的区别是什么,如何更改代码以消除此错误。

BTW,我在Windows 10上使用RStudio 0.99.903。

1 个答案:

答案 0 :(得分:2)

RMatrix的{​​{3}}标记为explicit

inline explicit RMatrix(const Source& source) 
   : data_(const_cast<Source&>(source).begin()),
     nrow_(source.nrow()),
     ncol_(source.ncol())
{}

您已为Par_MatVec_Mult构造函数提供此签名

Par_MatVec_Mult(RMatrix<double> A, vector<double> v, vector<double> Av)

并且稍后尝试将NumericMatrix传递给它。这将需要在RMatrix的构造中进行隐式转换,但由于本来是合适的构造函数被标记为explicit,因此不允许这样做,并且您会收到错误。

以下是一个简化示例:

#include <Rcpp.h>

class Wrapper {
private:
    std::size_t nr;
    std::size_t nc;

public:
    template <typename T>
    explicit Wrapper(const T& t)
        : nr(t.nrow()),
          nc(t.ncol())
    {}

    std::size_t size() const { return nr * nc; }
};

// [[Rcpp::export]]
int test(Rcpp::NumericMatrix x) {
    // Wrapper w = x;   // Error: conversion from ‘Rcpp::NumericMatrix
                        // {aka Rcpp::Matrix<14>}’ to non-scalar
                        // type ‘Wrapper’ requested

    Wrapper w(x);       // Ok

    return w.size();
}

在您的情况下,修复很简单:只需将Par_MatVec_Mult的构造函数的签名更改为:

Par_MatVec_Mult(NumericMatrix A, vector<double> v, vector<double> Av)