如何将向量传递给基于推力的odeint观察器的构造函数,以便可以在仿函数中读取它

时间:2014-08-29 18:42:06

标签: c++ boost thrust odeint

我正在使用boost的push与odeint一起扩展参数研究示例,我不知道如何将值向量传递给观察者的构造函数,以便可以从内部访问这些值(只读)观察者的函子。

以下是仅适用于观察者的代码。

//// Observes the system, comparing the current state to 
//// values in unchangingVector

struct minimum_perturbation_observer { 
  struct minPerturbFunctor
  {
    template< class T >
    __host__ __device__
    void operator()( T t ) const
    {
    //// I would like to be able to read any member 
    //// of m_unchangingVector here.
    }
  };


  // CONSTRUCTOR
  minimum_perturbation_observer( size_t N, state_type unchangingVector, int len) : 
        m_N( N ),
        m_output( N ),
        m_unchangingVector( len ) // len is the correct length of unchangingVector
  {
    // all trials start with output = 0
    thrust::fill( m_output.begin() , m_output.end() , 0.0 );

    // copy unchangingVector to m_unchangingVector, the latter 
    // of which should be accessible from the functor operator() 
    // above.
    thrust::copy( unchangingVector.begin(), unchangingVector.end(),
                  m_unchangingVector.begin());
  }

  template< class State >
  void operator()(State x , value_type t )
  {
    thrust::for_each(
                 thrust::make_zip_iterator( thrust::make_tuple(
                                   boost::begin( x ) + 0*m_N,
                                   boost::begin( x ) + 1*m_N,
                                   boost::begin( m_output )
                                   )
                            ),
                 thrust::make_zip_iterator( thrust::make_tuple(
                                   boost::begin( x ) + 1*m_N,
                                   boost::begin( x ) + 2*m_N,
                                   boost::begin( m_output ) + m_N
                                       )
                            ) ,
                 minPerturbFunctor() );
  }

  // variables
  size_t m_N; // number of trials (i.e. number of initial conditions)
  state_type m_output;   // of length N_ICS
  state_type m_unchangingVector; // 
};

我已尝试制作m_unchangingVector staticconst,但这不正确,因为需要在实例化观察者时进行设置。

或者,也许最好的方法是将unchangingVector作为thrust::make_zip_iterator( thrust::make_tuple(...中的另一个参数传递,但我觉得这些项会按照状态变量的方式被索引(这不是我想要的) )。一个可能有用的答案是解释函数声明中的(T t)含义,以及如何将unchangingVector作为同一个对象传递给评估运算符的每个线程。

我认为这可能只是选择正确的关键字变量描述符,但我不知道使用哪一个,我不知道如何查找/计算出来。

上面代码的错误是error: a nonstatic member reference must be relative to a specific object。当我尝试在仿函数中访问m_unchangingVector时抛出。


经过进一步的探索,我觉得我已经确定了完成这项任务的正确方法,但我仍然陷入困境。

我在构造函数中添加了一个构造函数。

  struct minPerturbFunctor
  {

    minPerturbFunctor( state_type unchangingVector, int len ) :
    f_unchangingVector( len ) 
    {
      // copy from argument to local vector (probably unnecessary, but
      // getting errors about calling host-functions from device/host 
      // so being paranoid about trying to make sure things are device-side
      thrust::copy( f_unchangingVector.begin(), 
                    f_unchangingVector.end(), 
                    unchangingVector.begin());
      f_len = len;      
    };

    template< class T >
    __host__ __device__
    void operator()( T t ) const
    {
        // I can now access f_len here (progress!)
        // But when I try to access any element via e.g., 
        // f_unchangingVector[0] I get the error below
    }
  };

警告:从主机 设备调用主机功能(“thrust :: detail :: vector_base&gt; :: operator []”) function(“minimum_perturbation_observer :: minPerturbFunctor :: operator()&gt;”)是不允许的

错误消息 /usr/local/cuda/bin/..//include/thrust/detail/function.h(104):错误:调用主机函数(“thrust :: device_vector&gt; :: device_vector “)来自设备功能(”thrust :: detail :: device_function :: device_function“)是不允许的

我做错了什么?

1 个答案:

答案 0 :(得分:4)

您可以将推力矢量传递给仿函数,但您无法轻松将其存储在此处。但是您可以将基础原始指针存储在此向量中:

struct minPerturbFunctor
{
    state_type::value_type* m_ptr;
    size_t m_len;
    minPerturbFunctor( state_type const& x )
    : m_ptr( thrust::raw_pointer_cast(&x[0]) )
    , m_len( x.size() )
    { }

    template< class T >
    __host__ __device__
    void operator()( T t ) const
    {
        // now you can access m_ptr like m_ptr[i] 
    }
};

这几乎是Robert Crovella的建议。