reduce_kernel_impl.h File Reference
#include <limits>
#include <TiledArray/external/cuda.h>
#include <thrust/device_vector.h>
#include <thrust/execution_policy.h>
#include <thrust/functional.h>
#include <thrust/reduce.h>
#include <thrust/transform_reduce.h>
Include dependency graph for reduce_kernel_impl.h:

Classes

struct  TiledArray::detail::absolute_value< T >
 

Namespaces

 TiledArray
 
 TiledArray::detail
 

Functions

template<typename T , typename ReduceOp >
TiledArray::reduce_cuda_kernel_impl (ReduceOp &&op, const T *arg, std::size_t n, T init, cudaStream_t stream, int device_id)
 T = reduce(T* arg) More...
 
template<typename T >
TiledArray::product_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id)
 
template<typename T >
TiledArray::sum_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id)
 
template<typename T >
TiledArray::max_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id)
 
template<typename T >
TiledArray::min_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id)
 
template<typename T >
TiledArray::absmax_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id)
 
template<typename T >
TiledArray::absmin_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id)