CuValue< Real > Class Template Reference

The following class is used to simulate non-const references to Real, e.g. More...

#include <cu-value.h>

Collaboration diagram for CuValue< Real >:

Public Member Functions
	CuValue (Real *data)
	CuValue (const CuValue &other)
CuValue	operator= (const CuValue< Real > &other)
Real	operator= (Real r)
Real	operator+= (Real r)
Real	operator-= (Real r)
	operator Real () const

Private Attributes
Real *	data_

Detailed Description

template<typename Real>
class kaldi::CuValue< Real >

The following class is used to simulate non-const references to Real, e.g.

as returned by the non-const operator (). This class is also used as a convenient way of reading a single Real value from the device.

Definition at line 34 of file cu-value.h.

Constructor & Destructor Documentation

CuValue() [1/2]

CuValue ( Real *  data )

inline

Definition at line 36 of file cu-value.h.

: data_(data) { }

CuValue() [2/2]

CuValue ( const CuValue< Real > &  other )

inline

Definition at line 37 of file cu-value.h.

: data_(other.data_) { }

Member Function Documentation

operator Real()

operator Real ( ) const

inline

Definition at line 73 of file cu-value.h.

References CuValue< Real >::data_.

                                { // assignment to Real
#if HAVE_CUDA == 1
  if (CuDevice::Instantiate().Enabled()) {
    Real value;
    CU_SAFE_CALL(cudaMemcpyAsync(&value, data_, sizeof(Real), 
                 cudaMemcpyDeviceToHost, cudaStreamPerThread));
    CU_SAFE_CALL(cudaStreamSynchronize(cudaStreamPerThread));
    return value;
  } else
#endif
    return *data_;
  }

operator+=()

Real operator+= ( Real  r )

inline

Definition at line 69 of file cu-value.h.

{ return (*this = r + Real(*this)); }

operator-=()

Real operator-= ( Real  r )

inline

Definition at line 70 of file cu-value.h.

{ return (*this = Real(*this) - r); }

operator=() [1/2]

CuValue operator= ( const CuValue< Real > &  other )

inline

Definition at line 39 of file cu-value.h.

References CuValue< Real >::data_.

                                                         {
#if HAVE_CUDA == 1
    if (CuDevice::Instantiate().Enabled()) {
      CU_SAFE_CALL(
        cudaMemcpyAsync(data_, other.data_, sizeof(Real), 
                        cudaMemcpyDeviceToDevice, cudaStreamPerThread));
      return *this;
    } else
#endif
    {
      *data_ = *other.data_;
      return *this;
    }
  }

operator=() [2/2]

Real operator= ( Real  r )

inline

Definition at line 54 of file cu-value.h.

References CuValue< Real >::data_.

                                  { // assignment from Real
#if HAVE_CUDA == 1
    if (CuDevice::Instantiate().Enabled()) {
      CU_SAFE_CALL(cudaMemcpyAsync(data_, &r, sizeof(Real), 
            cudaMemcpyHostToDevice, cudaStreamPerThread));
      CU_SAFE_CALL(cudaStreamSynchronize(cudaStreamPerThread));
      return r;
    } else
#endif
    {
      *data_ = r;
      return r;
    }
  }

Member Data Documentation

data_

Real* data_

private

Definition at line 86 of file cu-value.h.

Referenced by CuValue< Real >::operator Real(), and CuValue< Real >::operator=().

---

The documentation for this class was generated from the following file:

• cudamatrix/cu-value.h