#include <quda_constants.h>#include <float_vector.h>#include <comm_quda.h>#include <blas_quda.h>#include <cub/block/block_reduce.cuh>
Include dependency graph for cub_helper.cuh:
This graph shows which files directly or indirectly include this file:
Go to the source code of this file.
Classes | |
| struct | quda::vector_type< scalar, n > |
| struct | quda::ReduceArg< T > |
| struct | quda::reduce_vector< T > |
Namespaces | |
| quda | |
Functions | |
| template<typename scalar , int n> | |
| __device__ __host__ void | quda::zero (vector_type< scalar, n > &v) |
| template<typename scalar , int n> | |
| __device__ __host__ vector_type< scalar, n > | quda::operator+ (const vector_type< scalar, n > &a, const vector_type< scalar, n > &b) |
| template<int block_size_x, int block_size_y, typename T , bool do_sum = true, typename Reducer = cub::Sum> | |
| __device__ void | quda::reduce2d (ReduceArg< T > arg, const T &in, const int idx=0) |
| template<int block_size, typename T , bool do_sum = true, typename Reducer = cub::Sum> | |
| __device__ void | quda::reduce (ReduceArg< T > arg, const T &in, const int idx=0) |
| template<int block_size_x, int block_size_y, typename T > | |
| __device__ void | quda::reduceRow (ReduceArg< T > arg, const T &in) |
Variables | |
| __device__ unsigned int | quda::count [QUDA_MAX_MULTI_REDUCE] = { } |
| __shared__ bool | quda::isLastBlockDone |
| __shared__ volatile bool | quda::isLastWarpDone [16] |
Detailed Description
Description
Provides helper functors for custom datatypes for cub algorithms.
Definition in file cub_helper.cuh.
