Могу ли я использовать thrust::host_vector или я должен использовать cudaHostAlloc для нулевого копирования с помощью Thrust?

Question

Я хочу использовать нулевую копию в отображенной памяти с помощью cudaHostGetDevicePointer.Могу ли я использовать thrust::host_vector или я должен использовать cudaHostAlloc(...,cudaHostAllocMapped)? Или это как-то проще сделать с помощью Thrust?

Answer 1

Я почти уверен, что по-прежнему невозможно использовать thrust::host_vector в качестве сопоставленного выделения хоста.Существует закрепленный распределитель памяти, но я не верю, что отображенная память доступна.Что вам нужно сделать, это что-то вроде этого:

1. Выделенная сопоставленная, закрепленная память хоста с cudaHostAlloc
2. Получите указатель устройства для памяти с нулевой копией, используя cudaHostGetDevicePointer
3. Создать thrust::device_ptr используя thrust::device_pointer_cast на этом указателе устройства (см. здесь для получения дополнительной информации)

Вы можете либо сделать thrust::device_vector используя thrust::device_ptr или аккуратно передать thrust::device_ptr к любым алгоритмам, которые принимают итератор.