hipcub/util_allocator.hpp Source File
hipcub/util_allocator.hpp Source File
hipCUB: hipcub/util_allocator.hpp Source File
30 #ifndef HIPCUB_UTIL_ALLOCATOR_HPP_
31 #define HIPCUB_UTIL_ALLOCATOR_HPP_
33 #ifdef __HIP_PLATFORM_AMD__
34 #include "backend/rocprim/util_allocator.hpp"
35 #elif defined(__HIP_PLATFORM_NVIDIA__)
36 #include "backend/cub/util_allocator.hpp"