YAKL_ScalarLiveOut.h

Go to the documentation of this file.

 
#pragma once
// Included by YAKL.h
 
__YAKL_NAMESPACE_WRAPPER_BEGIN__
namespace yakl {
 
  template <class T> class ScalarLiveOut {
  protected:
    Array<T,1,memDevice,styleC> data;
 
  public:
    YAKL_INLINE ScalarLiveOut() {
      data = Array<T,1,memDevice,styleC>("ScalarLiveOut_data",1);  // Create array
    }
    explicit ScalarLiveOut(T val, Stream stream = Stream() ) {
      data = Array<T,1,memDevice,styleC>("ScalarLiveOut_data",1);  // Create array
      hostWrite(val,stream);                                // Copy to device
    }
    YAKL_INLINE ~ScalarLiveOut() {
      data = Array<T,1,memDevice,styleC>();
    }
 
    YAKL_INLINE ScalarLiveOut            ( ScalarLiveOut const &rhs) { this->data = rhs.data; }
    YAKL_INLINE ScalarLiveOut & operator=( ScalarLiveOut const &rhs) { this->data = rhs.data; return *this; }
    YAKL_INLINE ScalarLiveOut            ( ScalarLiveOut      &&rhs) { this->data = rhs.data; }
    YAKL_INLINE ScalarLiveOut & operator=( ScalarLiveOut      &&rhs) { this->data = rhs.data; return *this; }
 
    template <class TLOC , typename std::enable_if< std::is_arithmetic<TLOC>::value , int >::type = 0>
    YAKL_INLINE T &operator= (TLOC rhs) const { data(0) = rhs; return data(0); }
 
    YAKL_INLINE T &operator() () const {
      return data(0);
    }
 
    YAKL_INLINE T get() const {
      return data(0);
    }
 
    inline T hostRead(Stream stream = Stream()) const {
      return data.createHostCopy(stream)(0);
    }
 
    inline void hostWrite(T val, Stream stream = Stream()) {
      // Copy data to device
      auto &myData = this->data;
      c::parallel_for( c::Bounds<1>(1) , YAKL_LAMBDA (int dummy) {
        myData(0) = val;
      } , DefaultLaunchConfig().set_stream(stream) );
    }
 
  };
}
__YAKL_NAMESPACE_WRAPPER_END__