24 #ifndef SKA_CHEETAH_CUDA_UTILS_CUDA_THRUST_H 25 #define SKA_CHEETAH_CUDA_UTILS_CUDA_THRUST_H 27 #include "cheetah/version.h" 30 #pragma GCC diagnostic push 31 #pragma GCC diagnostic ignored "-Wall" 32 #pragma GCC diagnostic ignored "-Wpragmas" 33 #pragma GCC diagnostic ignored "-Wunused-parameter" 34 #pragma GCC diagnostic ignored "-Wunused-variable" 35 #include <thrust/device_vector.h> 36 #include <thrust/host_vector.h> 37 #include <thrust/device_ptr.h> 38 #include <thrust/device_malloc_allocator.h> 39 #include <thrust/transform.h> 40 #include <thrust/iterator/counting_iterator.h> 41 #include <thrust/iterator/transform_iterator.h> 42 #include <thrust/iterator/constant_iterator.h> 43 #include <thrust/iterator/zip_iterator.h> 44 #include <thrust/gather.h> 45 #include <thrust/functional.h> 46 #include <thrust/execution_policy.h> 47 #include <thrust/transform_reduce.h> 48 #include <thrust/for_each.h> 49 #include <thrust/complex.h> 50 #include <thrust/tuple.h> 51 #include <thrust/extrema.h> 52 #include <thrust/detail/raw_pointer_cast.h> 53 #include <thrust/reduce.h> 54 #include <thrust/sort.h> 55 #include <thrust/distance.h> 56 #include <thrust/copy.h> 57 #include <thrust/fill.h> 58 #include <thrust/execution_policy.h> 59 #include <thrust/system/cuda/execution_policy.h> 60 #include <thrust/system/cuda/vector.h> 61 #pragma GCC diagnostic pop 63 #if ((CHEETAH_CUDA_MAJOR_VERSION == 7) && (CHEETAH_CUDA_MINOR_VERSION == 5)) 64 #define CUDA_BROKEN_THRUST_WITH_ALLOCATOR 66 #if ((CHEETAH_CUDA_MAJOR_VERSION == 8) && (CHEETAH_CUDA_MINOR_VERSION == 0)) 67 #define CUDA_BROKEN_THRUST_WITH_ALLOCATOR 69 #endif // broken thrust 76 namespace cuda_utils {
82 #endif // SKA_CHEETAH_CUDA_UTILS_CUDA_THRUST_H Some limits and constants for FLDO.