•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
Search Results
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 > | |
T 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 > | |
T TiledArray::product_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id) | |
template<typename T > | |
T TiledArray::sum_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id) | |
template<typename T > | |
T TiledArray::max_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id) | |
template<typename T > | |
T TiledArray::min_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id) | |
template<typename T > | |
T TiledArray::absmax_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id) | |
template<typename T > | |
T TiledArray::absmin_reduce_cuda_kernel_impl (const T *arg, std::size_t n, cudaStream_t stream, int device_id) | |