YAKL_intrinsics_any.h

Go to the documentation of this file.

 
#pragma once
// Included by YAKL_intrinsics.h
 
__YAKL_NAMESPACE_WRAPPER_BEGIN__
namespace yakl {
  namespace intrinsics {
 
    template <class T, int rank, int myStyle>
    inline bool any( Array<T,rank,memHost,myStyle> arr ) {
      #ifdef YAKL_DEBUG
        if (!arr.initialized()) { yakl_throw("ERROR: calling any on an array that has not been initialized"); }
      #endif
      bool any_true = false;
      for (int i=0; i < arr.totElems(); i++) { if (arr.data()[i]) any_true = true; }
      return any_true;
    }
 
    template <class T, int rank, int myStyle>
    inline bool any( Array<T,rank,memDevice,myStyle> arr , Stream stream = Stream() ) {
      #ifdef YAKL_DEBUG
        if (!arr.initialized()) { yakl_throw("ERROR: calling any on an array that has not been initialized"); }
      #endif
      ScalarLiveOut<bool> any_true(false,stream);
      c::parallel_for( "YAKL_internal_any" , arr.totElems() , YAKL_LAMBDA (int i) { if (arr.data()[i]) any_true = true; },
                       DefaultLaunchConfig().set_stream(stream) );
      return any_true.hostRead(stream);
    }
 
    template <class T, int rank, unsigned D0, unsigned D1, unsigned D2, unsigned D3>
    inline bool any( SArray<T,rank,D0,D1,D2,D3> const &arr ) {
      bool any_true = false;
      for (int i=0; i < arr.totElems(); i++) { if (arr.data()[i]) any_true = true; }
      return any_true;
    }
 
    template <class T, int rank, class B0, class B1, class B2, class B3>
    inline bool any( FSArray<T,rank,B0,B1,B2,B3> const &arr ) {
      bool any_true = false;
      for (int i=0; i < arr.totElems(); i++) { if (arr.data()[i]) any_true = true; }
      return any_true;
    }
 
  }
}
__YAKL_NAMESPACE_WRAPPER_END__