9 #if defined(__clang__) && defined(__CUDA__) && CUDA_VERSION >= 9000 10 #define CUB_USE_COOPERATIVE_GROUPS 13 #include <thrust/system/cuda/vector.h> 14 #include <thrust/system/cuda/execution_policy.h> 15 #include <thrust/transform_reduce.h> 16 #include <thrust/device_ptr.h> 17 #include <thrust/device_vector.h> 18 #include <thrust/sort.h> 20 #define device_malloc(size) quda::device_malloc_(__func__, quda::file_name(__FILE__), __LINE__, size) 21 #define device_free(ptr) quda::device_free_(__func__, quda::file_name(__FILE__), __LINE__, ptr) void deallocate(char *ptr, size_t n)
#define pool_device_malloc(size)
char * allocate(std::ptrdiff_t num_bytes)
#define pool_device_free(ptr)