Как да избегнем конструкцията по подразбиране на елементи в thrust::device_vector?

thrust::device_vector конструира елементите, които съдържа, като използва предоставения си разпределител, точно както std::vector. Възможно е да се контролира какво прави разпределителят, когато векторът поиска от него да конструира елемент.

Използвайте персонализиран разпределител, за да избегнете инициализация по подразбиране на векторни елементи:

// uninitialized_allocator is an allocator which
// derives from device_allocator and which has a
// no-op construct member function
template<typename T>
  struct uninitialized_allocator
    : thrust::device_malloc_allocator<T>
{
  // note that construct is annotated as
  // a __host__ __device__ function
  __host__ __device__
  void construct(T *p)
  {
    // no-op
  }
};

// to make a device_vector which does not initialize its elements,
// use uninitialized_allocator as the 2nd template parameter
typedef thrust::device_vector<float, uninitialized_allocator<float> > uninitialized_vector;

Все още ще поемате разходите за стартиране на ядрото, за да извикате uninitialized_allocator::construct, но това ядро ​​ще бъде без операция и ще се оттегли бързо. Това, от което наистина се интересувате, е избягването на честотната лента на паметта, необходима за запълване на масива, което прави това решение.

Има пълен примерен код тук.

Имайте предвид, че тази техника изисква Thrust 1.7 или по-добра.