Annotate allocator_traits::allocate & allocator_traits::deallocate with __host__ __device__.