Могу ли я использовать итератор CUDA CUB вместо тяги? - PullRequest
Новая
       87

Могу ли я использовать итератор CUDA CUB вместо тяги?

0 голосов
/ 09 октября 2018

Можно ли использовать итераторы с CUB, такие как Thrust?Я хочу использовать CUB вместо толчка следующим образом: 

__global__ void reduce_roster(thrust::device_vector<float>::iterator vect, float * tab_seq, int SEUIL_ROSTER)
{
    int tid = blockIdx.x * blockDim.x + threadIdx.x;
    float resultat = 0;

    int i = TAILLE_ROSTER/TAILLE_SEQ_ROSTER;
    resultat = thrust::reduce(thrust::device, vect + (tid * TAILLE_ROSTER) + TAILLE_SEQ_ROSTER *  (i - 1), vect + (tid * TAILLE_ROSTER) + TAILLE_SEQ_ROSTER * i);


    float tmp;
    i--;

    while (resultat != -1 && i != 0)
    {
        if (resultat > SEUIL_ROSTER)
        {
            resultat = -1; 
        }
        else
        {               

            tmp = thrust::reduce(thrust::device, vect + (tid * TAILLE_ROSTER) + TAILLE_SEQ_ROSTER *  (i - 1), vect + (tid * TAILLE_ROSTER) + TAILLE_SEQ_ROSTER * i);

            resultat = resultat + tmp;


            i--;
        }
    }

Поскольку CUB работает быстрее, чем доверие, я пытаюсь использовать его для уменьшения сегментов массива.

1 Ответ

0 голосов
/ 10 октября 2018

Нет.

И, кроме того, невозможно использовать CUB для выполнения той же операции, что вы показали в своем коде.В зависимости от того, как вы компилируете ваше ядро, Thrust будет либо генерировать код, в котором сокращение выполняется последовательно каждым потоком, либо при запуске дочернего ядра с использованием динамического параллелизма.CUB, с другой стороны, имеет функции уменьшения масштаба устройства и деформации.Он не поддерживает однопотоковые операции или операции с областями динамического параллелизма.

...